Zoe Wu

Hi, I'm Zoe! I'm currently a Post-baccalaureate Researcher at the Kempner Institute at Harvard University.

My research is focused on developing algorithmic techniques that enable inference-time control over generative modeling. My research seeks to use the theoretical arguments of advanced statistical inference techniques to create empirically performant, principled algorithms with applications ranging across robotics, protein generation, reasoning, and computer vision.

Check out my latest project, STEER: diverse low-temperature diffusion sampling through replica exchange.

I am applying to PhD programs this cycle (Fall 2027 entry). I'm always happy to chat about research and potential opportunities.

Zoe Wu

Recent News

Jul 2025 – Present
Started as a Post-baccalaureate Researcher at the Kempner Institute at Harvard University!
Aug 2024 – Mar 2025
Student Researcher at Harvard University — Generative AI research with Professor Natesh Pillai.
May 2024 – Aug 2024
Student Researcher at Carnegie Mellon University — Computational Neuroscience with Professor Byron Yu.
Sep 2023 – May 2024
Student Researcher at Harvard University — Theoretical Statistics research with Professor Sitan Chen.
May 2023 – Aug 2023
Engineering Intern at Nuro in Mountain View, CA.
Feb 2023 – May 2023
Engineering Intern at Amira in Somerville, MA.
Jul 2022 – Dec 2022
Engineering Intern at iRobot in Bedford, MA.
Jul 2021 – Mar 2022
Student Researcher at the Wyss Institute for Biologically Inspired Engineering — Computer Vision with Professor Robert Wood.
Sep 2021 – Feb 2022
Project Manager for Onboarding Students, Harvard Undergraduate Robotics Club.

Research

Diverse Low-Temperature Diffusion Sampling through Replica Exchange figure
Diverse Low-Temperature Diffusion Sampling through Replica Exchange
Zoe Wu, Haldun Balim, Yilun Du
Diffusion models achieve strong generative performance across domains. However, achieving high sample quality under hard physical and dynamical constraints, as in protein design and robotic manipulation, remains difficult. Existing inference-time methods either require a costly or unavailable verifier, or avoid a verifier at the cost of reduced diversity. To relieve this tension, we introduce Score Tempering with Exchange between Exploratory Replicas (STEER), a verifier-free inference-time method that replaces single-chain sharpening with global search using multiple diffusion chains. During sampling, neighboring chains, each at different sharpening strengths, exchange particles according to an energy-based acceptance rate, propagating exploration to outputted samples. Across a diverse range of domains, from synthetic multimodal distributions to image generation, STEER achieves a better fidelity–diversity trade-off and improves success rates on continuous-control tasks in both verifier-based and verifier-free settings. Our findings suggest that effective diffusion tempering requires going beyond local updates to population-based global search that enhance fidelity and diversity.
Geometry-Aware Energy-Based Image Modeling figure
Geometry-Aware Energy-Based Image Modeling
Zoe Wu
AAAI CLIP Workshop, 2026
Recent advances in image generative models have enabled rich collaborations between humans and AI systems. Among these, Energy-Based Models (EBMs) learn an energy landscape that guides noised samples toward high-probability regions. Unlike diffusion models that use fixed time schedules, EBMs possess equilibrium properties that enable user feedback during generation without destabilizing the distribution. However, current EBM research primarily optimizes for high-fidelity images, offering little control over the trade-off between semantic realism and fine-grained diversity—an essential feature for interactive creative applications. Artists and creatives thus lack a modality to balance semantic coherence (e.g., “a red apple”) with creative variation (e.g., apples of different shapes or colors). To address this, we introduce a geometry-aware annealing framework for EBMs. We propose a directionally-aware annealing variable that leverages local geometric information to directionally adjust the effective noise level during sampling.
PCA trajectories of M1 and PFC population activity in visual and memory trials
Neural Basis of Visual and Memory Guided Reaching in Prefrontal and Motor Cortex
Zoe Wu, Adithya Narayan Chandrasekaran, Byron Yu
Recorded from prefrontal (PFC) and motor (M1) cortex during delayed visually and memory guided reaches to study how each region encodes sensory input and motor plans. Population decoding shows PFC shifting from processing sensory input to maintaining a motor plan, while M1 maintains the motor plan in both visual and memory trials.