About Me

I am Yifan Zhang, currently pursuing my Ph.D. in Machine Learning at the Institute for Interdisciplinary Information Sciences (IIIS) at Tsinghua University. I have the distinct honor of being advised by Prof. Yang Yuan and Prof. Andrew Chi-Chih Yao. Previously, I had the privilege of undertaking my undergraduate studies at Yuanpei College of Peking University.

I am open to potential cooperation or visiting opportunities, feel free to contact me via email, X, or WeChat!

Research Topics

• AI Reasoning
• Pretraining & Alignment
• Representation Learning

Selected Publications

• Augmenting Math Word Problems via Iterative Question Composing [code]
  Haoxiong Liu*, Yifan Zhang*, Yifan Luo, Andrew Chi-Chih Yao.
  International Conference on Learning Representations (DPFM @ ICLR 2024)

• Autonomous Data Selection with Language Models for Mathematical Texts [code]
  Yifan Zhang*, Yifan Luo*, Yang Yuan, Andrew Chi-Chih Yao.
  International Conference on Learning Representations (DPFM @ ICLR 2024)
  • Featured as Huggingface Daily Papers: https://huggingface.co/papers/2402.07625
  • Huggingface dataset: https://huggingface.co/datasets/math-ai/AutoMathText

• Meta Prompting for AI Systems [code]
  Yifan Zhang, Yang Yuan, Andrew Chi-Chih Yao.
  International Conference on Learning Representations (BGPT @ ICLR 2024)

• Cumulative Reasoning with Large Language Models [code]
  Yifan Zhang*, Jingqin Yang*, Yang Yuan, Andrew Chi-Chih Yao.
  International Conference on Learning Representations (BGPT @ ICLR 2024)

• Information Flow in Self-Supervised Learning [code]
  Zhiquan Tan, Jingqin Yang, Weiran Huang, Yang Yuan†, Yifan Zhang†.
  International Conference on Machine Learning (ICML 2024)

• Matrix Information Theory for Self-Supervised Learning [code]
  Yifan Zhang*, Zhiquan Tan*, Jingqin Yang*, Weiran Huang, Yang Yuan.
  International Conference on Machine Learning (ICML 2024)

• Contrastive Learning Is Spectral Clustering On Similarity Graph [code]
  Zhiquan Tan*, Yifan Zhang*, Jingqin Yang*, Yang Yuan.
  International Conference on Learning Representations (ICLR 2024)

• Trade-off Between Efficiency and Consistency for Removal-based Explanations [code]
  Yifan Zhang*, Haowei He*, Zhiquan Tan, Yang Yuan.
  Conference on Neural Information Processing Systems (NeurIPS 2023)

( * denotes equal contribution, † denotes corresponding authors )