Bio

I am a PhD candidate at the UC Santa Cruz working with Prof. Yi Zhang. I received my Bachelor's degree in Computer Science with a Statistics minor, and Master's degree in Natural Language Processing from UC Santa Cruz.

My research aims to empower AI systems with advanced reasoning capabilities. I work at the intersection of the following topics:
  • Complex Planning and Reasoning: My research aims to solve real-world problems, which often require multi-step reasoning and actions.
  • Multi-agent Collaboration: I work on MAS (multi-agent systems), so my study also includes how agents coordinate, communicate, and share information.
  • Information Retrieval: I conduct research on, but not limited to, RAG with LLM; broadly speaking, I care about Retrieval-Augmented AI.

News

Experience

  • Google DeepMind: Student Researcher, 2025
  • Amazon Alexa: Applied Scientist Intern, 2023
  • Mineral.ai, an Alphabet company, previous Google[X] project: AI Resident, 2023, 2024

Teaching and Mentoring

Instructor:
  As of 2025 Fall, I am serving as the graduate student instructor for NLP 255 – Topics in Applied Natural Language Processing.

Research Mentor:
  I am dedicated to mentoring students, ranging from high schoolers to Grad students, supporting them in pursuing their career goals in either academic research or engineering in AI.

List of Mentored Students
Linda Zeng (2023-Present) The Harker School
Owen Arnst (2024-Present) UCSC BS
Chung-Hsiang Lo (2025) Northeastern University MS
Rithwik Gupta (2023-2025) Irvington High School
Divij Motwani (2023-2025) Palo Alto High School → UC Berkeley
Jimmy Lu (2024-2025) UCSC → UIUC
Timothy Wei (2024) Saratoga High School → UIUC
Hsien Xin Peng (2024) Ridge High School

Teaching assistant:
  2023, 2024 - Applied Machine Learning: Deep Learning
  2023 - Advanced Topics in Natural Language Processing
  2022 - Programming Abstractions

Services

Reviewer:
  2026 - AAAI, ECIR
  2025 - ECIR, ICLR, ICML, SIGIR, ACL, NeurIPS, Knowledge-Based Systems
  2024 - TheWebConf (WWW), CVPR, ICML, NeurIPS
  2023 - NeurIPS

Event Organizer:
  Co-organized the tools-meet-llm tutorial at ACM SIGIR 2024.
  Co-chair and organizer of the 1st AI Student Research Symposium at UC Santa Cruz.

Selected Publications

Full publication list can be found on Google Scholar page. (* equal contribution, † co-advisor).

Publication Image
GenIR: Generative Visual Feedback for Mental Image Retrieval
Diji Yang*, Minghao Liu*, Chung-Hsiang Lo, Yi Zhang, James Davis
NeurIPS 2025 [Paper] [Code]

Publication Image
Worse than Zero-shot? A Fact-Checking Dataset for Evaluating the Robustness of RAG Against Misleading Retrievals
Linda Zeng, Rithwik Gupta, Divij Motwani, Diji Yang** , Yi Zhang**
NeurIPS 2025 [Paper] [Dataset]

Publication Image
Knowing You Don't Know: Learning When to Continue Search in Multi-round RAG through Self-Practicing
Diji Yang*, Linda Zeng*, Jinmeng Rao, Yi Zhang
SIGIR 2025 [Paper] [Code]

Publication Image
Right this way: Can VLMs Guide Us to See More to Answer Questions?
Li Liu*, Diji Yang*, Sijia Zhong, Kalyana Suma Sree Tholeti, Lei Ding, Yi Zhang, Leilani H. Gilpin
NeurIPS 2024 [Paper] [Code] [Dataset]

Publication Image
IM-RAG: Multi-Round Retrieval-Augmented Generation Through Learning Inner Monologues
Diji Yang, Jinmeng Rao, Kezhen Chen, Xiaoyuan Guo, Yawen Zhang, Jie Yang, Yi Zhang
SIGIR 2024 [Paper]

Publication Image
Tackling vision language tasks through learning inner monologues
Diji Yang, Jinmeng Rao, Kezhen Chen, Xiaoyuan Guo, Yawen Zhang, Jie Yang, Yi Zhang
AAAI 2024 [Paper]

Get Connected

I am always interested in working with new collaborators from all backgrounds on research ideas and/or impactful projects related to my areas of interest. That being said, I commit the sufficient time to each project I take on, so my availability for new projects may depends on current commitments.

If you’re interested in collaborating, feel free to send me an email to find a time to chat!