Jiayi Zhang
Other people with similar names: Jiayi Zhang
Unverified author pages with similar names: Jiayi Zhang
2026
Concise Math Reasoning via Difficulty-Aware Distillation
Yifan Wu | Jingze Shi | Bingheng Wu | Jiayi Zhang | Xiaotian Lin | Yizhang Zhu | Zhaoyang Yu | Bang Liu | Chenglin Wu | Nan Tang | Yuyu Luo
Findings of the Association for Computational Linguistics: ACL 2026
Yifan Wu | Jingze Shi | Bingheng Wu | Jiayi Zhang | Xiaotian Lin | Yizhang Zhu | Zhaoyang Yu | Bang Liu | Chenglin Wu | Nan Tang | Yuyu Luo
Findings of the Association for Computational Linguistics: ACL 2026
Human experts tackle difficult math problems by identifying and executing a few pivotal steps rather than listing every intermediate thought. In contrast, standard Chain-of-Thought (CoT) distillation trains small models on lengthy reasoning traces, encouraging a uniform overthinking style across easy and hard items alike. The result is rigid, slow solutions that sacrifice adaptivity. This approach stands in sharp contrast to human intuition. Humans naturally adapt their problem-solving strategy, dedicating significant effort to difficult problems while finding quick, simple solutions for easier ones. We argue that the root cause lies in the training data: it contains excess information and reasoning steps organized in ways misaligned with human practice. We address this with Difficulty-Aware Distillation(DAD), a procedure for producing training data that mirrors concise human reasoning. A large teacher model first assesses a problem’s difficulty and then rewrites the solution to retain only the essential steps. Using this process, we constructed LiteCoT, a 100,000-example corpus of short, clear rationales, and used it to train our Liter models. With 100k LiteCoT, we outperform models trained on 800k long CoT and cut both training and inference costs. The advantage is consistent across standard math benchmarks, showing that concise, human-aligned data delivers equal or better accuracy with much less compute. For example, on the challenging AIME24 exam, our approach reaches 74.2% Pass@1 using only about 5K inference tokens, surpassing other methods that consume many more tokens.
2025
Self-Supervised Prompt Optimization
Jinyu Xiang | Jiayi Zhang | Zhaoyang Yu | Xinbing Liang | Fengwei Teng | Jinhao Tu | Fashen Ren | Xiangru Tang | Sirui Hong | Chenglin Wu | Yuyu Luo
Findings of the Association for Computational Linguistics: EMNLP 2025
Jinyu Xiang | Jiayi Zhang | Zhaoyang Yu | Xinbing Liang | Fengwei Teng | Jinhao Tu | Fashen Ren | Xiangru Tang | Sirui Hong | Chenglin Wu | Yuyu Luo
Findings of the Association for Computational Linguistics: EMNLP 2025
Well-designed prompts are crucial for enhancing Large language models’ (LLMs) reasoning capabilities while aligning their outputs with task requirements across diverse domains. However, manually designed prompts require expertise and iterative experimentation. While existing prompt optimization methods aim to automate this process, they rely heavily on external references such as ground truth or by humans, limiting their applicability in real-world scenarios where such data is unavailable or costly to obtain. To address this, we propose Self-Supervised Prompt Optimization (SPO), a cost-efficient framework that discovers effective prompts for both closed and open-ended tasks without requiring external reference. Motivated by the observations that prompt quality manifests directly in LLM outputs and LLMs can effectively assess adherence to task requirements, we derive evaluation and optimization signals purely from output comparisons. Specifically, SPO selects superior prompts through pairwise output comparisons evaluated by an LLM evaluator, followed by an LLM optimizer that aligns outputs with task requirements. Extensive experiments demonstrate that SPO outperforms state-of-the-art prompt optimization methods, achieving comparable or superior results with significantly lower costs (e.g., 1.1% to 5.6% of existing methods) and fewer samples (e.g., three samples).