← Search

Zijun Gao

11 accepted papers

2026

CORE: Concept-Oriented Reinforcement for Bridging the Definition–Application Gap in Mathematical Reasoning

ICLR 2026poster

Large language models (LLMs) often solve drill-style math exercises yet fail to apply the concept right when the problem requires genuine understanding. Popular outcome-based RL pipelines reinforce final answers but provide little fine-grained conceptual signal, so models improve at pattern reuse ra…

Cited by 0SourceScholar
2026

Partial Identification under High-Dimensional Potential Outcomes and Confounders via Optimal Transport

ICML 2026poster

Partial identification provides informative causal guarantees when point identification is impossible, but existing approaches based on optimal transport (OT) become computationally and statistically intractable in high-dimensional settings. This limitation is particularly severe when both potential…

Cited by 0SourceScholar
2025

Bridging Multiple Worlds: Multi-marginal Optimal Transport for Causal Partial-identification Problem

AISTATS 2025poster

Under the prevalent potential outcome model in causal inference, each unit is associated with multiple potential outcomes but at most one of which is observed, leading to many causal quantities being only partially identified. The inherent missing data issue echoes the multi-marginal optimal transpo…

Cited by 0SourceScholar
2025

SAGEPhos: Sage Bio-Coupled and Augmented Fusion for Phosphorylation Site Detection

ICLR 2025poster

Phosphorylation site prediction based on kinase-substrate interaction plays a vital role in understanding cellular signaling pathways and disease mechanisms. Computational methods for this task can be categorized into kinase-family-focused and individual kinase-targeted approaches. Individual kinase…

2025

Tightening Causal Bounds via Covariate-Aware Optimal Transport

ICML 2025poster

Causal estimands can vary significantly depending on the relationship between outcomes in treatment and control groups, leading to wide partial identification (PI) intervals that impede decision making. Incorporating covariates can substantially tighten these bounds, but requires determining the ran…

2024

Gradient-based Parameter Selection for Efficient Fine-Tuning

CVPR 2024poster

With the growing size of pre-trained models full fine-tuning and storing all the parameters for various downstream tasks is costly and infeasible. In this paper we propose a new parameter-efficient fine-tuning method Gradient-based Parameter Selection (GPS) demonstrating that only tuning a few selec…

2024

Images Speak Louder than Words: Understanding and Mitigating Bias in Vision-Language Model from a Causal Mediation Perspective

EMNLP 2024main

Vision-language models (VLMs) pre-trained on extensive datasets can inadvertently learn biases by correlating gender information with specific objects or scenarios. Current methods, which focus on modifying inputs and monitoring changes in the model’s output probability scores, often struggle to com…

Cited by 0SourcePDFScholar