← Search

Zaiyi Zheng

5 accepted papers

2025

Beyond the Permutation Symmetry of Transformers: The Role of Rotation for Model Fusion

ICML 2025spotlight

Symmetry in the parameter space of deep neural networks (DNNs) has proven beneficial for various deep learning applications. A well-known example is the permutation symmetry in Multi-Layer Perceptrons (MLPs), where permuting the rows of weight matrices in one layer and applying the inverse permutati…

2025

CoRAG: Enhancing Hybrid Retrieval-Augmented Generation through a Cooperative Retriever Architecture

EMNLP 2025

Retrieval-Augmented Generation (RAG) is introduced to enhance Large Language Models (LLMs) by integrating external knowledge. However, conventional RAG approaches treat retrieved documents as independent units, often overlooking their interdependencies. Hybrid-RAG, a recently proposed paradigm that

2025

Hierarchical Demonstration Order Optimization for Many-shot In-Context Learning

NeurIPS 2025poster

In-Context Learning (ICL) is a technique where large language models (LLMs) leverage multiple demonstrations (i.e., examples) to perform tasks. With the recent expansion of LLM context windows, many-shot ICL (generally with more than 50 demonstrations) can lead to significant performance improvement…

Cited by 0SourcecodeScholar
2025

SemCoT: Accelerating Chain-of-Thought Reasoning through Semantically-Aligned Implicit Tokens

NeurIPS 2025poster

Chain-of-Thought (CoT) enhances the performance of Large Language Models (LLMs) on reasoning tasks by encouraging step-by-step solutions. However, the verbosity of CoT reasoning hinders its mass deployment in efficiency-critical applications. Recently, implicit CoT approaches have emerged, which enc…

Cited by 0SourcecodeScholar
2024

Explaining Graph Neural Networks with Large Language Models: A Counterfactual Perspective on Molecule Graphs

EMNLP 2024finding

In recent years, Graph Neural Networks (GNNs) have become successful in molecular property prediction tasks such as toxicity analysis. However, due to the black-box nature of GNNs, their outputs can be concerning in high-stakes decision-making scenarios, e.g., drug discovery. Facing such an issue, G…