Yuzhong Chen
2026
Demystify the Role of Memory in Machine Learning Engineering Agents
Xinyu Zhao | Junpeng Wang | Yuzhong Chen | Menghai Pan | Chin-Chia Michael Yeh | Jiarui Sun | Yan Zheng | Mahashweta Das | Tianlong Chen
Findings of the Association for Computational Linguistics: ACL 2026
Xinyu Zhao | Junpeng Wang | Yuzhong Chen | Menghai Pan | Chin-Chia Michael Yeh | Jiarui Sun | Yan Zheng | Mahashweta Das | Tianlong Chen
Findings of the Association for Computational Linguistics: ACL 2026
While memory is a core component in agent systems, its behavioral impact in complex, long-horizon domains like machine learning engineering (MLE) remains poorly understood. Unlike short, reactive exchanges, MLE agents solve tasks through cycles of experimentation and improvement where past errors can inform future success. This paper presents a systematic study dissecting how memory influences agent behavior and performance across diverse MLE challenges. We first introduce a dynamic coding memory designed to capture and reuse debugging experiences, and integrate it into two representative agent paradigms: a sequential, chain-based agent that mirrors human-like iterative refinement, and a parallel, tree-based agent that performs broad, self-exploratory search in the code space. Our central finding is that the role of memory is contingent on the agent’s underlying architecture. For chain-based agents, memory proves highly beneficial, enabling them to avoid recurring mistakes and engage in more coherent, iterative refinement, which significantly improves reliability and task success. In contrast, for tree-based search agents, memory introduces a critical trade-off: it enhances procedural stability at the cost of constraining search diversity, which can prematurely narrow exploration and lead to suboptimal final solutions. These findings reveal a fundamental trade-off between procedural reliability and solution innovation modulated by memory, offering insights for designing more effective and robust MLE agents.
Feedback to Reasoning: LLM-Assisted Molecular Optimization with Domain Feedback and Historical Reasoning
Wenhan Gao | Xiran Fan | Chin-Chia Michael Yeh | Jiarui Sun | Yuzhong Chen | Menghai Pan | Mahashweta Das | Yi Liu
Findings of the Association for Computational Linguistics: ACL 2026
Wenhan Gao | Xiran Fan | Chin-Chia Michael Yeh | Jiarui Sun | Yuzhong Chen | Menghai Pan | Mahashweta Das | Yi Liu
Findings of the Association for Computational Linguistics: ACL 2026
The success of large language models (LLMs) across domains highlights their potential in scientific tasks, with molecular optimization being a promising frontier. Traditionally, this optimization relies on iterative expert feedback to refine molecules toward desired properties, a process well aligned with LLMs’ strengths. **As an experience-driven task, molecular optimization depends critically on the domain feedback and accumulation of historical knowledge. However, none of the existing methods fully leverages such feedback and historical knowledge with reasoning traces and chemical insights.** In this work, we propose F2R: Feedback to Reasoning, a conversational molecular optimization pipeline that enables LLMs to accumulate and retrieve past actions, rationales, and feedback. Like humans, LLMs can generate imperfect reasoning; F2R is the first framework to use detailed domain feedback to critique and improve this reasoning. This transforms LLMs from passive text generators into agentic experts that learn both actions and reasoning from experience. Consequently, F2R shows remarkable performance.
2025
Enhancing Foundation Models in Transaction Understanding with LLM-based Sentence Embeddings
Xiran Fan | Zhimeng Jiang | Chin-Chia Michael Yeh | Yuzhong Chen | Yingtong Dou | Menghai Pan | Yan Zheng
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing: Industry Track
Xiran Fan | Zhimeng Jiang | Chin-Chia Michael Yeh | Yuzhong Chen | Yingtong Dou | Menghai Pan | Yan Zheng
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing: Industry Track
The ubiquity of payment networks generates vast transactional data encoding rich consumer and merchant behavioral patterns. Recent foundation models for transaction analysis process tabular data sequentially but rely on index-based representations for categorical merchant fields, causing substantial semantic information loss by converting rich textual data into discrete tokens. While Large Language Models (LLMs) can address this limitation through superior semantic understanding, their computational overhead challenges real-time financial deployment. We introduce a hybrid framework that uses LLM-generated embeddings as semantic initializations for lightweight transaction models, balancing interpretability with operational efficiency. Our approach employs multi-source data fusion to enrich merchant categorical fields and a one-word constraint principle for consistent embedding generation across LLM architectures. We systematically address data quality through noise filtering and context-aware enrichment. Experiments on large-scale transaction datasets demonstrate significant performance improvements across multiple transaction understanding tasks.
2024
Enhancing Hyperbolic Knowledge Graph Embeddings via Lorentz Transformations
Xiran Fan | Minghua Xu | Huiyuan Chen | Yuzhong Chen | Mahashweta Das | Hao Yang
Findings of the Association for Computational Linguistics: ACL 2024
Xiran Fan | Minghua Xu | Huiyuan Chen | Yuzhong Chen | Mahashweta Das | Hao Yang
Findings of the Association for Computational Linguistics: ACL 2024
Knowledge Graph Embedding (KGE) is a powerful technique for predicting missing links in Knowledge Graphs (KGs) by learning the entities and relations. Hyperbolic space has emerged as a promising embedding space for KGs due to its ability to represent hierarchical data. Nevertheless, most existing hyperbolic KGE methods rely on tangent approximation and are not fully hyperbolic, resulting in distortions and inaccuracies. To overcome this limitation, we propose LorentzKG, a fully hyperbolic KGE method that represents entities as points in the Lorentz model and represents relations as the intrinsic transformation—the Lorentz transformations between entities. We demonstrate that the Lorentz transformation, which can be decomposed into Lorentz rotation/reflection and Lorentz boost, captures various types of relations including hierarchical structures. Experimental results show that our LorentzKG achieves state-of-the-art performance.