Publications

• Nudging the Boundaries of LLM Reasoning
  Justin Chih-Yao Chen, Becky Xiangyu Peng, Prafulla Kumar Choubey, Kung-Hsiang Huang, Jiaxin Zhang, Mohit Bansal, Chien-Sheng Wu
  Arxiv Preprint 2025 [paper] [Long]

• Symbolic Mixture-of-Experts: Adaptive Skill-based Routing for Heterogeneous Reasoning
  Justin Chih-Yao Chen, Sukwon Yun, Elias Stengel-Eskin, Tianlong Chen, Mohit Bansal
  Arxiv Preprint 2025 [paper] [code] [project page] [Long]

• MAgICoRe: Multi-Agent, Iterative, Coarse-to-Fine Refinement for Reasoning
  Justin Chih-Yao Chen, Archiki Prasad, Swarnadeep Saha, Elias Stengel-eskin and Mohit Bansal
  EMNLP 2025 [paper] [code] [Long]

• Learning to Generate Unit Tests for Automated Debugging
  Archiki Prasad, Elias Stengel-Eskin, Justin Chih-Yao Chen, Zaid Khan, Mohit Bansal
  COLM 2025 [paper] [code] [Long]

• Reverse Thinking Makes LLMs Stronger Reasoners
  Justin Chih-Yao Chen, Zifeng Wang, Hamid Palangi, Rujun Han, Sayna Ebrahimi, Long Le, Vincent Perot, Swaroop Mishra, Mohit Bansal, Chen-Yu Lee, Tomas Pfister
  NAACL 2025 [paper] [code] [Long]

• System-1.x: Learning to Balance Fast and Slow Planning with Language Models
  Swarnadeep Saha, Archiki Prasad, Justin Chih-Yao Chen, Peter Hase, Elias Stengel-Eskin, Mohit Bansal
  ICLR 2025 [paper] [code] [Long]

• ReConcile: Round-Table Conference Improves Reasoning via Consensus among Diverse LLMs
  Justin Chih-Yao Chen, Swarnadeep Saha, and Mohit Bansal
  ACL 2024 [paper] [code] [Long]

• MAGDi: Structured Distillation of Multi-Agent Interaction Graphs Improves Reasoning in Smaller Language Models
  Justin Chih-Yao Chen*, Swarnadeep Saha*, Elias Stengel-Eskin, Mohit Bansal
  ICML 2024 [paper] [code] [Long]

• Dual Graph Networks with Synthetic Oversampling for Imbalanced Rumor Detection on Social Media
  Yen-Wen Lu, Chih-Yao Chen, and Cheng-Te Li
  WWW 2024 [paper] [Short]

• Location-Aware Visual Question Generation with Lightweight Models
  Nicholas Collin Suwono, Justin Chih-Yao Chen, Tun Min Hung, Ting-Hao Kenneth Huang, I-Bin Liao, Yung-Hui Li, Lun-Wei Ku, Shao-Hua Sun
  EMNLP 2023 [paper] [Long]

• Label-Aware Hyperbolic Embeddings for Fine-grained Emotion Classification
  Chih-Yao Chen, Tun-Min Hung, Yi-Li Hsu, Lun-Wei Ku
  ACL 2023 [paper] [code] [Long]

• HonestBait: Forward References for Attractive but Faithful Headline Generation
  Chih-Yao Chen*, Dennis Wu*, Lun-Wei Ku
  ACL Findings 2023 [paper] [Long]

• ZS-BERT: Towards Zero-Shot Relation Extraction with Attribute Representation Learning
  Chih-Yao Chen and Cheng-Te Li
  NAACL 2021 [paper] [code] [Long]