← Search

Wanjia Zhao

11 accepted papers

2026

OpenThoughts: Data Recipes for Reasoning Models

ICLR 2026oral

Reasoning models have made rapid progress on many benchmarks involving math, code, and science. Yet, there are still many open questions about the best train- ing recipes for reasoning since state-of-the-art models often rely on proprietary datasets with little to no public information available. To…

Cited by 0SourcecodeScholar
2026

PRISM-Physics: Causal DAG-Based Process Evaluation for Physics Reasoning

ICLR 2026poster

Benchmarks for competition-style reasoning have advanced evaluation in mathematics and programming, yet physics remains comparatively underexplored. Most existing physics benchmarks evaluate only final answers, which fail to capture reasoning processes, while recent stepwise methods rely on heuristi…

Cited by 0SourceScholar
2026

Sample Lottery: Unsupervised Discovery of Critical Instances for LLM Reasoning

ICLR 2026poster

Reinforcement Learning with Verifiable Reward (RLVR) has equipped large language models (LLMs) with the capability of reasoning over complicated logical problems through policy optimization. However, conventional methods require complete annotation of the entire dataset and allocate computation unif…

Cited by 0SourceScholar
2025

DeepSeek-Prover-V1.5: Harnessing Proof Assistant Feedback for Reinforcement Learning and Monte-Carlo Tree Search

ICLR 2025poster

Lean is an advanced proof assistant designed to facilitate formal theorem proving by providing a variety of interactive feedback. In this paper, we explore methodologies to leverage proof assistant feedback to augment the capabilities of large language models in constructing formal proofs. First, we…

2025

Don’t Forget the Enjoin: FocalLoRA for Instruction Hierarchical Alignment in Large Language Models

NeurIPS 2025poster

Recent studies reveal that large language models (LLMs) often struggle to resolve conflicting instructions embedded within hierarchical prompts, resulting in decreased compliance with system-level directives and compromising the reliability of safety-critical applications. While earlier approaches a…

Cited by 0SourceScholar
2025

GeoAda: Efficiently Finetune Geometric Diffusion Models with Equivariant Adapters

NeurIPS 2025poster

Geometric diffusion models have shown remarkable success in molecular dynamics and structure generation. However, efficiently fine-tuning them for downstream tasks with varying geometric controls remains underexplored. In this work, we propose an SE(3)-equivariant adapter framework (GeoAda) that ena…

Cited by 0SourceScholar
2025

Protein Large Language Models: A Comprehensive Survey

EMNLP 2025

Protein-specific large language models (ProteinLLMs) are revolutionizing protein science by enabling more efficient protein structure prediction, function annotation, and design. While existing surveys focus on specific aspects or applications, this work provides the first comprehensive overview of

2025

Rethink GraphODE Generalization within Coupled Dynamical System

ICML 2025spotlight

Coupled dynamical systems govern essential phenomena across physics, biology, and engineering, where components interact through complex dependencies. While Graph Ordinary Differential Equations (GraphODE) offer a powerful framework to model these systems, their **generalization** capabilities degra…

Cited by 0SourcePDFScholar
2025

SiriuS: Self-improving Multi-agent Systems via Bootstrapped Reasoning

NeurIPS 2025poster

Multi-agent AI systems powered by large language models (LLMs) are increasingly applied to solve complex tasks. However, these systems often rely on fragile, manually designed prompts and heuristics, making optimization difficult. A key challenge in optimizing multi-agent systems is acquiring suitab…

Cited by 0SourcecodeScholar
2024

Physics-Informed Regularization for Domain-Agnostic Dynamical System Modeling

NeurIPS 2024poster

Learning complex physical dynamics purely from data is challenging due to the intrinsic properties of systems to be satisfied. Incorporating physics-informed priors, such as in Hamiltonian Neural Networks (HNNs), achieves high-precision modeling for energy-conservative systems. However, real-world s…

2023

Positive Distribution Pollution: Rethinking Positive Unlabeled Learning from a Unified Perspective

AAAI 2023technical

Positive Unlabeled (PU) learning, which has a wide range of applications, is becoming increasingly prevalent. However, it suffers from problems such as data imbalance, selection bias, and prior agnostic in real scenarios. Existing studies focus on addressing part of these problems, which fail to pro…

Cited by 4SourcePDFScholar