Fang Wu
Author directory2026
AgentFold: Closed-Loop Agentic Search for Protein Folding Model Design
Mingquan Liu | Jiangyu Chen | Hanqun Cao | Xujun Zhang | Pengsen Ma | Xiangru Tang | Shuting Jin | Annie Zheng | Zhuo Yang | Tianfan Fu | Fang Wu | Xiangxiang Zeng
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Mingquan Liu | Jiangyu Chen | Hanqun Cao | Xujun Zhang | Pengsen Ma | Xiangru Tang | Shuting Jin | Annie Zheng | Zhuo Yang | Tianfan Fu | Fang Wu | Xiangxiang Zeng
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Scientific LLM agents have shown promise in literature reasoning, tool use, and experiment planning, but it remains unclear whether they can autonomously improve large, tightly coupled scientific machine-learning systems through executable code changes and computationally expensive validation. We study this question in protein folding, where progress requires coordinated architectural modifications, multi-objective evaluation, and domain-aware interpretation. We present AgentFold, a multi-agent framework that formulates folding-model development as a closed-loop search over executable code variants. Starting from ESMFold, AgentFold proposes hypotheses, implements and debugs code-level modifications, evaluates model variants, analyzes experimental outcomes, and stores both successful and failed interventions in structured memory. An MCTS-style policy allocates computational resources across high-scoring search branches. On an engineering-scale protein-folding codebase comprising more than 2,000 lines of code, AgentFold explores approximately 80 model variants using approximately 5,000 GPU-hours and 170 million LLM tokens. Under a matched computational budget, AgentFold improves the best lDDT by 7.5% over independent Codex proposals and outperforms a random-search control. Beyond model improvement, the resulting intervention traces reveal recurring empirical design patterns: stable gains tend to arise from early, soft, learnable priors and gated refinement, whereas direct geometric perturbations and geometry-conditioned feedback often destabilize training. The code and experimental resources are publicly available at https://github.com/lmqfly/AgentFold.
W2T: LoRA Weights Already Know What They Can Do
Xiaolong Han | Ferrante Neri | Jiang Zijian | Fang Wu | Yanfang Ye | Lu Yin | Zehong Wang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Xiaolong Han | Ferrante Neri | Jiang Zijian | Fang Wu | Yanfang Ye | Lu Yin | Zehong Wang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Each LoRA checkpoint compactly stores task-specific updates in low-rank weight matrices, offering an efficient way to adapt large language models to new tasks and domains. In principle, these weights already encode what the adapter does and how well it performs. In this paper, we ask whether this information can be read directly from the weights, without running the base model or accessing training data. A key obstacle is that a single LoRA update can be factorized in infinitely many ways. Without resolving this ambiguity, models trained on the factors may fit the particular factorization rather than the underlying update. To this end, we propose W2T (Weight-to-Token), which maps each LoRA update to a provably canonical form via QR decomposition followed by SVD, so that all equivalent factorizations share the same representation. The resulting components are then tokenized and processed by a Transformer to produce a weight-space embedding. Across language and vision LoRA collections, W2T achieves strong results on attribute classification, performance prediction, and adapter retrieval, demonstrating that LoRA weights reliably indicate model behavior once factorization ambiguity is removed.
PRKV: Page Restructed KV Cache for High Accuracy and Efficiency LLM Generation
Fang Wu | Congming Gao | Weixi Zhu | Jiwu Shu
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Fang Wu | Congming Gao | Weixi Zhu | Jiwu Shu
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
As the key-value(KV) cache size scales with context length, accessing large KV cache each step and substantial GPU memory demand challenge us to deploy LLMs with long contexts.Various sparse attention methods have been proposed and offloading-based KV retrieval preserves entire KV cache in CPU memory and dynamically retrieves most relevant KV pairs for each decoding step, which performs higher quality and effectively reduces GPU memory consumption than other line works. However, exiting KV retrieval performs page-level to reduce estimation overhead, which introduces inaccurate KV selection and significant re- trieval overhead. We propose PRKV, a framework that both-optimizes algorithm and system for page-level KV retrieval with KV offloading. On the algorithm side, PRKV introduces hybrid KV selection that combines both static and dynamic KV selection strategies. On the system side, PRKV employs contiguous memory in- dexing and batched transfer optimizations to improve retrieval efficiency. Exper- iments demonstrate that PRKV improve accuracy across various scenarios and models, delivering up to 6.75× speedup compared to SOTA KV retrieval methods.
2025
AnchorAttention: Difference-Aware Sparse Attention with Stripe Granularity
Yu Zhang | Dong Guo | Fang Wu | Guoliang Zhu | Dian Ding | Yiming Zhang
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing
Yu Zhang | Dong Guo | Fang Wu | Guoliang Zhu | Dian Ding | Yiming Zhang
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing
Large Language Models (LLMs) with extended context lengths face significant computational challenges during the pre-filling phase, primarily due to the quadratic complexity of self-attention. Existing methods typically employ dynamic pattern matching and block-sparse low-level implementations. However, their reliance on local information for pattern identification fails to capture global contexts, and the coarse granularity of blocks leads to persistent internal sparsity, resulting in suboptimal accuracy and efficiency. To address these limitations, we propose AnchorAttention, a difference-aware, dynamic sparse attention mechanism that efficiently identifies critical attention regions at a finer stripe granularity while adapting to global contextual information, achieving superior speed and accuracy. AnchorAttention comprises three key components: (1) Pattern-based Anchor Computation, leveraging the commonalities present across all inputs to rapidly compute a set of near-maximum scores as anchor; (2) Difference-aware Stripe Sparsity Identification, performing difference-aware comparisons with anchor to quickly obtain discrete coordinates of significant regions in a stripe-like sparsity pattern; (3) Fine-grained Sparse Computation, replacing the traditional contiguous loading strategy with a discrete key-value loading approach to maximize sparsity rates while preserving hardware computational potential. Additionally, we integrate the identification strategy into a single operator to maximize parallelization potential. With its finer-grained sparsity strategy, AnchorAttention achieves higher sparsity rates at the same recall level, significantly reducing computation time. Compared to previous state-of-the-art methods, at a text length of 128k, it achieves a speedup of 1.44× while maintaining higher recall rates.
LocAgent: Graph-Guided LLM Agents for Code Localization
Zhaoling Chen | Robert Tang | Gangda Deng | Fang Wu | Jialong Wu | Zhiwei Jiang | Viktor Prasanna | Arman Cohan | Xingyao Wang
Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Zhaoling Chen | Robert Tang | Gangda Deng | Fang Wu | Jialong Wu | Zhiwei Jiang | Viktor Prasanna | Arman Cohan | Xingyao Wang
Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Code localization–identifying precisely where in a codebase changes need to be made–is a fundamental yet challenging task in software maintenance. Existing approaches struggle to efficiently navigate complex codebases when identifying relevant code snippets.The challenge lies in bridging natural language problem descriptions with the target code elements, often requiring reasoning across hierarchical structures and multiple dependencies.We introduce LocAgent, a framework that addresses code localization through a graph-guided agent.By parsing codebases into directed heterogeneous graphs, LocAgent creates a lightweight representation that captures code structures and their dependencies, enabling LLM agents to effectively search and locate relevant entities through powerful multi-hop reasoning.Experimental results on real-world benchmarks demonstrate that our approach significantly enhances accuracy in code localization.Notably, our method with the fine-tuned Qwen-2.5-Coder-Instruct-32B model achieves comparable results to SOTA proprietary models at greatly reduced cost (approximately 86% reduction), reaching up to 92.7% accuracy on file-level localization while improving downstream GitHub issue resolution success rates by 12% for multiple attempts (Pass@10). Our code is available at https://github.com/gersteinlab/LocAgent.
Large Language Models are Good Relational Learners
Fang Wu | Vijay Prakash Dwivedi | Jure Leskovec
Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Fang Wu | Vijay Prakash Dwivedi | Jure Leskovec
Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Large language models (LLMs) have demonstrated remarkable capabilities across various domains, yet their application to relational deep learning (RDL) remains underexplored. Existing approaches adapt LLMs by traversing relational links between entities in a database and converting the structured data into flat text documents, but this text-based serialization disregards critical relational structures, introduces redundancy, and often exceeds standard LLM context lengths. We introduce Rel-LLM, a novel architecture that employs a graph neural network (GNN) based encoder to create structured relational prompts for LLMs within a retrieval-augmented generation (RAG) framework. Unlike traditional text-based serialization approaches, our method preserves the inherent relational structure of databases while enabling LLMs to effectively process and reason over complex entity relationships. Specifically, the GNN encoder extracts a local subgraph around an entity to build feature representations that contain relevant entity relationships and temporal dependencies. These representations are transformed into structured prompts using a denormalization process, effectively allowing the LLM to reason over relational structures. Through extensive experiments, we demonstrate that Rel-LLM outperforms existing methods on key RDL tasks, offering a scalable and efficient approach to integrating LLMs with structured data sources. Code is available at https://github.com/smiles724/Rel-LLM.
2024
InsertGNN: A Hierarchical Graph Neural Network for the TOEFL Sentence Insertion Problem
Fang Wu | Stan Z. Li
Findings of the Association for Computational Linguistics: EMNLP 2024
Fang Wu | Stan Z. Li
Findings of the Association for Computational Linguistics: EMNLP 2024
The integration of sentences poses an intriguing challenge within the realm of NLP, but it has not garnered the attention it deserves. Existing methods that focus on sentence arrangement, textual consistency, and question answering have been shown to be inadequate in addressing this issue. To bridge this gap, we introduce InsertGNN which conceptualizes the problem as a graph and employ a hierarchical Graph Neural Network (GNN) to comprehend the interplay between sentences. Our approach was rigorously evaluated on a TOEFL dataset, and its efficacy was further validated on the expansive arXiv dataset using cross-domain learning. Thorough experimentation unequivocally establishes InsertGNN’s superiority over all comparative benchmarks, achieving an impressive 70% accuracy—a performance on par with average human test scores.
Search
Fix author
Co-authors
- Xiangru Tang 2
- Hanqun Cao 1
- Jiangyu Chen 1
- Zhaoling Chen 1
- Arman Cohan 1
- Gangda Deng 1
- Dian Ding 1
- Vijay Prakash Dwivedi 1
- Tianfan Fu 1
- Congming Gao 1
- Dong Guo 1
- Xiaolong Han 1
- Zhiwei Jiang 1
- Shuting Jin 1
- Jure Leskovec 1
- Stan Z. Li 1
- Mingquan Liu 1
- Pengsen Ma 1
- Ferrante Neri 1
- Viktor Prasanna 1
- Jiwu Shu 1
- Xingyao Wang 1
- Zehong Wang 1
- Jialong Wu 1
- Zhuo Yang 1
- Yanfang Ye 1
- Lu Yin 1
- Xiangxiang Zeng 1
- Xujun Zhang 1
- Yiming Zhang 1
- Yu Zhang 1
- Annie Zheng 1
- Guoliang Zhu 1
- Weixi Zhu 1
- Jiang Zijian 1