Jiaxin Zhang

Jiaxin Zhang

Research Lead
Salesforce AI Research

Email: jxzhangai@gmail.com

I am a Senior Staff Research Scientist and Research Lead at Salesforce AI Research. I study how to build trustworthy and self-improving AI agents, with a focus on safety, alignment, and reliability. My work combines agentic post-training, behavioral evaluation, and model–harness co-evolution to connect honesty, calibration, and uncertainty with learning and control in coding, tool-using, and long-horizon agents.

I am interested in the gap between what an agent knows, what it does, and when it should stop or ask for help. Addressing this gap requires more than a stronger model: training signals, evaluation protocols, the execution harness, and the environment all shape agent behavior.

Previously, I was a Senior Staff Research Scientist and founding research lead at Intuit AI Research, where I built LLM reliability and evaluation infrastructure used by 1,600+ internal users, and automated prompt optimization pipelines used by 2,000+ developers. This work was recognized with the Intuit CTO Award (top 1%). Earlier, at Oak Ridge National Laboratory, I worked on distributed deep learning at 20,000+ GPUs on Summit and served as PI/co-PI on seven DOE/ORNL projects ($6.4M total) in AI for Science. I received the DOE Promising Early-Career Researcher Award and earned my Ph.D. at Johns Hopkins University.

Research Directions

News & Updates


09/2026 Prospective Hindsight: Self-Calibrating Reinforcement Learning via Prediction–Reality Gaps has been accepted to NeurIPS 2026.
08/2026 Agentic Uncertainty Quantification and Seeing is Believing? Evaluating Vision-Language Model Susceptibility in Agent-to-Agent Multimodal Persuasion have been accepted to EMNLP 2026.
05/2026 2 papers accepted by ICML 2026: Agentic Confidence Calibration and LaTtE-Flow.
04/2026 Presenting our NuRL paper (code) at ICLR 2026 in Brazil! 🇧🇷
04/2026 We release CaOPD — calibration-aware on-policy distillation! Read the paper, check out the code and Hugging Face.
04/2026 2 papers accepted by ACL 2026: From Passive Metric to Active Signal and Don’t Stop Early: Scalable Enterprise Deep Research.
02/2026 2 invited talks on Building Reliable Long-horizon Agents at UCSD EnCORE Workshop and EPFL.
01/2026 1 paper accepted by ICLR 2026: Nudging the Boundaries of LLM Reasoning.
12/2025 Attending NeurIPS 2025 in San Diego! 🇺🇸
10/2025 We published a blog on Towards Trustworthy Enterprise Deep Research at Salesforce.

Selected Publications


For the full publication list, see Google Scholar.

01 Safety, Alignment & Agent Reliability

Behavioral alignment, honesty, uncertainty, calibration, and reliable decision-making—from LLMs to agent trajectories.

  1. ICML 2026
    Jiaxin Zhang, Caiming Xiong, and Chien-Sheng Wu
    In International Conference on Machine Learning, 2026

    Process-level signals for estimating whether an agent trajectory will succeed.

  2. EMNLP 2026
    Jiaxin Zhang, Prafulla Kumar Choubey, Kung-Hsiang Huang, Caiming Xiong, and Chien-Sheng Wu
    In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, 2026

    Turning uncertainty into active memory and reflection controls for agents.

  3. ACL 2026
    Jiaxin Zhang, Wendi Cui, Zhuohang Li, Lifu Huang, Bradley Malin, Caiming Xiong, and Chien-Sheng Wu
    In Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics, 2026

    A survey of uncertainty as a control and training signal, rather than only a diagnostic metric.

  4. EMNLP 2026
    Haoyi Qiu, Yilun Zhou, Pranav Narayanan Venkit, Kung-Hsiang Huang, Jiaxin Zhang, Nanyun Peng, and Chien-Sheng Wu
    In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, 2026

    Evaluating behavioral susceptibility to agent-to-agent multimodal persuasion.

  5. Under review
    Agentic Self-Awareness: Aligning Knowing, Acting, and Holding
    Qinglin Chen, Hiroaki Hayashi, Jiaxin Zhang, and Chien-Sheng Wu
    2026
    Submitted to ICLR 2027.

    Testing the gap between recognizing blocked tasks and actually withholding unauthorized or infeasible actions.

  6. EMNLP 2025
    Zhuohang Li, Chao Yan, Nicholas J Jackson, Wendi Cui, Bo Li, Jiaxin Zhang, and Bradley A Malin
    In Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing, 2025

    Statistical approaches to factuality guarantees for vision-language models.

  7. NAACL 2025
    Yu Wang, Kamalika Das, Xiang Gao, Wendi Cui, Peng Li, and Jiaxin Zhang
    In Proceedings of the 2025 Conference of the North American Chapter of the Association for Computational Linguistics, 2025

    Inference-time attention editing to mitigate contextual hallucinations.

  8. EACL 2024
    Xiang Gao, Jiaxin Zhang, Lalla Mouatadid, and Kamalika Das
    In Proceedings of the 18th Conference of the European Chapter of the Association for Computational Linguistics, 2024

    Quantifying language-model uncertainty through input perturbations.

  9. AISTATS 2024
    Jiaxin Zhang, Kamalika Das, and Sricharan Kumar
    In International Conference on Artificial Intelligence and Statistics, 2024

    Earlier UQ foundations: distance-aware representations for deterministic uncertainty estimation.

  10. EMNLP 2023
    Jiaxin Zhang, Zhuohang Li, Kamalika Das, Bradley Malin, and Sricharan Kumar
    In Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing, 2023

    Black-box hallucination detection through semantic-aware cross-check consistency.

  11. CVPR 2022
    Zhuohang Li, Jiaxin Zhang, Luyang Liu, and Jian Liu
    In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2022

    Earlier privacy foundations: auditing federated-learning defenses through gradient leakage.

02 Agentic Post-training & Reinforcement Learning

Reinforcement learning, on-policy distillation, experience-based supervision, and cost-effective adaptation of language models.

  1. NeurIPS 2026
    Jiaxin Zhang, Xiangyu Peng, Qinglin Chen, Yu Li, Hiroaki Hayashi, and Chien-Sheng Wu
    In Advances in Neural Information Processing Systems, 2026
    Accepted September 24, 2026.

    Self-calibrating reinforcement learning using the gap between predicted and observed outcomes.

  2. arXiv 2026
    Jiaxin Zhang, Xiangyu Peng, Qinglin Chen, Qinyuan Ye, Caiming Xiong, and Chien-Sheng Wu
    2026
    Under review at ICLR 2027.

    Separating capability gains from confidence calibration in on-policy distillation.

  3. ICLR 2026
    Justin Chih-Yao Chen, Becky Xiangyu Peng, Prafulla Kumar Choubey, Kung-Hsiang Huang, Jiaxin Zhang, Mohit Bansal, and Chien-Sheng Wu
    In International Conference on Learning Representations, 2026

    Hint-guided exploration to help reinforcement learning move beyond all-failure rollouts.

  4. Under review
    Self-Retrospection Distillation: Turning Post-hoc Experiences into Prior Foresight
    Haoxiang Zhang, Qinglin Chen, Hiroaki Hayashi, Zhuofeng Li, Siming Zhang, Jiaxin Zhang, Jixuan Chen, Fang Wu, Pan Lu, Silvio Savarese, Julian McAuley, and Chien-Sheng Wu
    2026
    Submitted to ICLR 2027.

    Distilling post-hoc experience into prior foresight, without requiring the privileged experience at inference time.

  5. Under review
    Fork Where the Model Changes Its Mind: Belief-Shift Branching for Tree-Structured Reinforcement Learning
    Bin Lei, Yu Li, Prafulla Kumar Choubey, Jiaxin Zhang, Xiangyu Peng, Qinyuan Ye, Kartik Narayan, Caiwen Ding, Silvio Savarese, and Chien-Sheng Wu
    2026
    Submitted to ICLR 2027.

    Allocating tree-structured exploration to points where the model’s answer belief changes.

  6. Under review
    Lessons from Agentic Post-Training of MLLMs for Multimodal Web Search
    Kartik Narayan, Xiangyu Peng, Prafulla Kumar Choubey, Jiaxin Zhang, Bin Lei, Vishal M. Patel, Silvio Savarese, and Chien-Sheng Wu
    2026
    Submitted to ICLR 2027.

    Studying how rollout design, tools, summarizers, and judges affect agentic post-training and evaluation.

  7. EMNLP 2025
    Jiaxin Zhang
    In Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing: Industry Track, 2025

    Confidence-aware control of self-guided reasoning trajectories.

  8. EMNLP 2024
    Jiaxin Zhang, Wendi Cui, Yiran Huang, Kamalika Das, and Sricharan Kumar
    In Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing, 2024

    Synthesizing structured knowledge for injection through continual pre-training, SFT, and retrieval.

  9. NeurIPS 2023
    Jiaxin Zhang, Zhuohang Li, Kamalika Das, and Sricharan Kumar
    In Advances in Neural Information Processing Systems, 2023

    Cost-effective language-model adaptation through sparse human supervision and multi-fidelity feedback.

03 Self-Improving Agents & Model–Harness Co-Evolution

Joint improvement of models, harnesses, and training data, alongside earlier work on automated prompt optimization.

  1. Under review
    MDS-Zero: Self-Improving Multimodal Deep Search via Online QA–Solver Co-training
    Xiangyu Peng, Jiaxin Zhang, An Yan, Yu Li, Ran Xu, Zeyuan Chen, and Chien-Sheng Wu
    2026
    Submitted to ICLR 2027.

    Co-training a question generator and search solver with a fixed verification harness and seeded training data.

  2. Under review
    Qinyuan Ye, Yu Li, Yada Pruksachatkun, Jiaxin Zhang, and Chien-Sheng Wu
    2026
    Submitted to ICLR 2027.

    Examining how randomness, task order, and underspecified memory updates undermine reproducible self-improvement.

  3. Under review
    CoTrace: Data Recipes for Training Terminal Agents with Harness–Model Co-Evolution
    Jixuan Chen, Jiaxin Zhang, Qinyuan Ye, Yada Pruksachatkun, Haoxiang Zhang, Jingming Zhuo, Yifan Zhang, Yutong Dai, Juntao Tan, Xiangyu Peng, Silvio Savarese, Zeyuan Chen, Lianhui Qin, and Chien-Sheng Wu
    2026
    Submitted to ICLR 2027.

    Routing failures into harness changes, runtime-matched training traces, and fresh online RL rollouts.

  4. ACL 2025
    Wendi Cui, Zhuohang Li, Hao Sun, Damien Lopez, Kamalika Das, Bradley A Malin, Sricharan Kumar, and Jiaxin Zhang
    In Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics, 2025

    Earlier prompt-level optimization: balancing strategic exploration and exploitation.

  5. ACL 2025
    Wendi Cui, Zhuohang Li, Hao Sun, Damien Lopez, Kamalika Das, Bradley A Malin, Sricharan Kumar, and Jiaxin Zhang
    In Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics, 2025

    A survey of heuristic-search methods for automated prompt optimization.

  6. EMNLP 2024
    Oral
    Wendi Cui, Zhuohang Li, Damien Lopez, Kamalika Das, Bradley Malin, Sricharan Kumar, and Jiaxin Zhang
    In Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing: Industry Track, 2024

    Decomposed consistency evaluation coupled with explanation-guided response improvement.

  7. EMNLP 2024
    Ankita Sinha, Wendi Cui, Kamalika Das, and Jiaxin Zhang
    In Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing - Industry Track, 2024

    Jointly optimizing prompt performance and security rather than performance alone.

04 Agent & Multimodal Systems

Long-horizon search, retrieval, memory, multimodal understanding and generation, and system evaluation and efficiency.

  1. ACL 2026
    Prafulla Kumar Choubey, Kung-Hsiang Huang, Pranav Narayanan Venkit, Jiaxin Zhang, Vaibhav Vats, Yu Li, Xiangyu Peng, and Chien-Sheng Wu
    In Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics: Industry Track, 2026

    Dependency-controlled information flow and evidence-aware stopping for enterprise deep research.

  2. ICML 2026
    Ying Shen, Zhiyang Xu, Jiuhai Chen, Shizhe Diao, Jiaxin Zhang, Yuguang Yao, Joy Rimchala, Ismini Lourentzou, and Lifu Huang
    In International Conference on Machine Learning, 2026

    Unifying image understanding and generation with layerwise timestep experts for efficient flow-based sampling.

  3. Under review
    Speculate with Memory: Lossless Acceleration for LLM Agents
    Yu Li, Qinyuan Ye, Prafulla Kumar Choubey, Jiaxin Zhang, and Chien-Sheng Wu
    2026
    Submitted to ICLR 2027.

    Using memory to improve a speculative executor, with exact-match acceptance and read-only tool constraints.

  4. EMNLP 2025
    Oral
    Kaijie Chen, Zihao Lin, Zhiyang Xu, Ying Shen, Yuguang Yao, Joy Rimchala, Jiaxin Zhang, and Lifu Huang
    In Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing, 2025

    A benchmark for reasoning-driven text-to-image generation.

  5. ICLR 2025
    Zhiyang Xu, Minqian Liu, Ying Shen, Joy Rimchala, Jiaxin Zhang, Qifan Wang, Yu Cheng, and Lifu Huang
    In International Conference on Learning Representations, 2025

    Modality-specialized adaptation layers and interleaved instruction data for joint text-and-image understanding and generation.

  6. EMNLP 2024
    Oral
    Zhuohang Li, Jiaxin Zhang, Chao Yan, Kamalika Das, Sricharan Kumar, Murat Kantarcioglu, and Bradley A Malin
    In Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing, 2024

    Testing whether a retrieval corpus can actually support a query or query distribution.

  7. EMNLP 2024
    Weichao Zhou, Jiaxin Zhang, Hilaf Hasson, Anu Singh, and Wenchao Li
    In Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing, 2024

    Efficient retrieval-context ranking with hypothetical query embeddings.

  8. EMNLP 2024
    Minqian Liu, Zhiyang Xu, Zihao Lin, Trevor Ashby, Joy Rimchala, Jiaxin Zhang, and Lifu Huang
    In Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing, 2024

    Reference-free evaluation of interleaved text-and-image generation.

Recent Talks


→ all talks

Awards & Honors


  • • Intuit CTO Award (Top 1% Performance), Intuit 2024
  • • Intuit A2D Innovation Award (Top 1%, Team Lead), Intuit 2024, 2025
  • • Promising Early-Career Researcher Award, Oak Ridge National Laboratory, US Department of Energy 2020
  • • Chinese Outstanding Students Abroad Award, Ministry of Education of the P.R. China 2019
  • • Acheson J. Duncan Graduate Research Award, Johns Hopkins University 2018
  • • Dean's Fellowship, Johns Hopkins University 2014
  • • China National Scholarship, Ministry of Education of the P.R. China 2009, 2012

Professional Services


  • Area Chair: NeurIPS, ICLR, ACL, EMNLP, NAACL 2024–now
  • Reviewer: NeurIPS, ICML, ICLR, ACL, EMNLP, NAACL, TMLR, JMLR, CVPR, ICCV, ECCV, AAAI, AISTATS, KDD 2020–now

Conference Travel


  • • Jul 2026, ICML 2026 @ Seoul 🇰🇷
  • • Apr 2026, ICLR 2026 @ Rio de Janeiro 🇧🇷
  • • Dec 2025, NeurIPS 2025 @ San Diego 🇺🇸
  • • Dec 2024, NeurIPS 2024 @ Vancouver 🇨🇦
  • • Nov 2024, EMNLP 2024 @ Miami 🇺🇸
  • • Jul 2024, ICML 2024 @ Vienna 🇦🇹
  • • May 2024, AISTATS 2024 @ Valencia 🇪🇸
  • • Jan 2024, WACV 2024 @ Hawaii 🇺🇸
  • • Dec 2023, NeurIPS 2023 @ New Orleans 🇺🇸
  • • Dec 2023, EMNLP 2023 @ Singapore 🇸🇬
  • • Feb 2023, AAAI 2023 @ Washington, DC 🇺🇸
  • • Dec 2022, NeurIPS 2022 @ New Orleans 🇺🇸
  • • Jul 2022, ICML 2022 @ Baltimore 🇺🇸
  • • Jun 2022, CVPR 2022 @ New Orleans 🇺🇸
  • • Dec 2021, NeurIPS 2021 @ Online 🌐
  • • Dec 2020, NeurIPS 2020 @ Online 🌐
  • • Dec 2019, NeurIPS 2019 @ Vancouver 🇨🇦