avatar

Yufeng Wu

Computer Science
Williams College '24
sw20 (at) williams.edu
yfwu02 (at) gmail.com


About Me

I graduated from Williams College in June, 2024 with a B.A. with Honors in Computer Science and Economics.

My research mainly lies in the intersection of Machine Learning and Causal Inference. I wrote my undergraduate thesis with Prof. Rohit Bhattacharya on developing novel causal inference methods in social networks where data is non-i.i.d. (non-independent and identically distributed). My thesis can be accessed here.

In summer 2023, I participated in the SMALL NSF REU program and collaborated with CSAFE to improve the robustness and explainability of machine learning models for shoeprint matching in forensic applications. I was fortunate to receive direct guidance from Profs. Xizhen Cai, Anna Plantinga, and Elizabeth Upton during this experience.

Research Interests

I’m interted in AI reliability, including robustness, safety, and generalizability. I’d like to investigate why models fail in certain scenarios and create methods to help models perform reliably across diverse real-world contexts.

News

See more details (e.g. photos, slides) here.

Papers

  1. Network Causal Effect Estimation In Graphical Models Of Contagion And Latent Confounding
    Yufeng Wu, Rohit Bhattacharya
    Causal Learning and Reasoning (CLeaR), 2025
    PDF Code Accepted
    Oral Presentation at American Causal Inference Conference (ACIC) 2024

  2. Improving and Evaluating Machine Learning Methods for Forensic Shoeprint Matching
    Divij Jain*, Saatvik Kher*, Lena Liang*, Yufeng Wu*, Ashley Zheng*, Xizhen Cai, Anna Plantinga, Elizabeth Upton (* = equal contribution)
    Journal of the Royal Statistical Society Series C: Applied Statistics
    PDF Code Project Page Under Review

Honors and Awards