Dawei Zhu
Author directoryOther people with similar names: Dawei Zhu
Unverified author pages with similar names: Dawei Zhu
2026
PsyPath: Psychologically-guided Self-Exploration for Personality Detection
Zheng Li | Hongxin Ding | Chenyu Zhang | Weimin Xiong | Dawei Zhu | Sujian Li
Findings of the Association for Computational Linguistics: ACL 2026
Zheng Li | Hongxin Ding | Chenyu Zhang | Weimin Xiong | Dawei Zhu | Sujian Li
Findings of the Association for Computational Linguistics: ACL 2026
Personality detection aims to label an individual’s traits via identifying linguistic cues from his or her written text. Previous approaches typically perform a direct mapping between text and trait labels or apply static reasoning to this task.In this paper, we argue that dynamic reasoning, underpinned by psychological theory, is essential for personality trait inference. To address this, we propose PsyPath, a novel framework that models personality detection as a process of psychologically-guided self-exploration. By enabling large language models (LLMs) to dynamically generate and answer psychologically meaningful questions, our method creates a dynamic reasoning path to explore the underlying dimensions of personality traits. This mechanism not only makes the reasoning process transparent, but also helps the model understand personality nuances in a way that mirrors expert psychological reasoning.For the “guided self-exploration”, we propose a novel hybrid scoring mechanism to step-by-step evaluate the generated nodes in the reasoning paths that balances psychological coherence (black-box scoring) and model output dynamics (white-box scoring). This reasoning-based formulation inherently reflects how psychologists assess personality, as they rely on iterative, diagnostic reasoning. Experiments on two benchmark datasets demonstrate that PsyPath consistently outperforms strong baselines, yielding improvements in predictive accuracy and model interpretability.Moreover, the generated reasoning paths provide psychologically meaningful training data, significantly improving performance and psychologically grounded interpretability in downstream tasks.
DocLens: A Tool-Augmented Multi-Agent Framework for Long Visual Document Understanding
Dawei Zhu | Rui Meng | Jiefeng Chen | Sujian Li | Tomas Pfister | Jinsung Yoon
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Dawei Zhu | Rui Meng | Jiefeng Chen | Sujian Li | Tomas Pfister | Jinsung Yoon
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Comprehending long visual documents, where information is distributed across extensive pages of text and visual elements, is a critical but challenging task for modern Vision-Language Models (VLMs). Existing approaches falter on a fundamental challenge: evidence localization. They struggle to retrieve relevant pages and overlook fine-grained details within visual elements, leading to limited performance and model hallucination. To address this, we propose DocLens, a tool-augmented multi-agent framework that effectively “zooms in” on evidence like a lens. It first navigates from the full document to specific visual elements on relevant pages, then employs a sampling-adjudication mechanism to generate a single, reliable answer. Paired with Gemini-2.5-Pro, DocLens achieves state-of-the-art performance on MMLongBench-Doc and FinRAGBench-V, surpassing even human experts. The framework’s superiority is particularly evident on vision-centric and unanswerable queries, demonstrating the power of its enhanced localization capabilities.
2025
More Tokens, Lower Precision: Towards the Optimal Token-Precision Trade-off in KV Cache Compression
Jiebin Zhang | Dawei Zhu | Yifan Song | Wenhao Wu | Chuqiao Kuang | Xiaoguang Li | Lifeng Shang | Qun Liu | Sujian Li
Findings of the Association for Computational Linguistics: EMNLP 2025
Jiebin Zhang | Dawei Zhu | Yifan Song | Wenhao Wu | Chuqiao Kuang | Xiaoguang Li | Lifeng Shang | Qun Liu | Sujian Li
Findings of the Association for Computational Linguistics: EMNLP 2025
As large language models (LLMs) process increasing context windows, the memory usage of KV cache has become a critical bottleneck during inference. The mainstream KV compression methods, including KV pruning and KV quantization, primarily focus on either token or precision dimensions separately. However, these works have left the trade-off between these two orthogonal dimensions largely unexplored. In this paper, we leverage the Information Bottleneck principle to formulate KV cache compression within a unified theoretical framework. We demonstrate that a carefully managed token-precision trade-off can achieve an optimal point within the Information Bottleneck compared to standalone KV pruning or KV quantization. Experiments reveal that storing more tokens in the KV cache at lower precision—a strategy we term quantized pruning—can significantly enhance the long-context performance of LLMs. An in-depth analysis of this token-precision trade-off across key aspects shows that quantized pruning achieves substantial improvements in retrieval-related tasks and consistently performs well across varying input lengths. Furthermore, quantized pruning exhibits notable stability and effectiveness across different KV pruning methods, quantization strategies, and model scales. These findings offer valuable insights into optimizing KV cache compression through balanced token-precision trade-off strategies. Our code is available at https://github.com/zhzihao/QPruningKV.
Chain-of-Thought Matters: Improving Long-Context Language Models with Reasoning Path Supervision
Dawei Zhu | Xiyu Wei | Guangxiang Zhao | Wenhao Wu | Haosheng Zou | Junfeng Ran | XWang | Lin Sun | Xiangzheng Zhang | Sujian Li
Findings of the Association for Computational Linguistics: EMNLP 2025
Dawei Zhu | Xiyu Wei | Guangxiang Zhao | Wenhao Wu | Haosheng Zou | Junfeng Ran | XWang | Lin Sun | Xiangzheng Zhang | Sujian Li
Findings of the Association for Computational Linguistics: EMNLP 2025
Recent advances in Large Language Models (LLMs) have highlighted the challenge of handling long-context tasks, where models need to reason over extensive input contexts to aggregate target information. While Chain-of-Thought (CoT) prompting has shown promise for multi-step reasoning, its effectiveness for long-context scenarios remains underexplored. Through systematic investigation across diverse tasks, we demonstrate that CoT’s benefits generalize across most long-context scenarios and amplify with increasing context length. Motivated by this, we propose a process-supervised framework that teaches models to generate high-quality reasoning paths for enhanced long-context performance. Our framework incorporates a self-sampling mechanism to bootstrap reasoning paths and a novel quality assessment protocol specifically designed for long-context scenarios. This protocol evaluates both answer correctness and process reliability, with the latter decomposed into source faithfulness and intrinsic consistency components for efficient and accurate assessment. Experimental results on various long-context benchmarks demonstrate the effectiveness of our approach, achieving significant improvements over outcome supervision baselines on both in-domain tasks (+13.6/+3.8 points for LLaMA/Qwen on MuSiQue) and cross-domain generalization (+9.3/+8.1 points on average across diverse QA tasks). Our code, data and trained models will be released upon acceptance.
LongAttn: Selecting Long-context Training Data via Token-level Attention
Longyun Wu | Dawei Zhu | Guangxiang Zhao | Zhuocheng Yu | Junfeng Ran | Xiangyu Wong | Lin Sun | Sujian Li
Findings of the Association for Computational Linguistics: ACL 2025
Longyun Wu | Dawei Zhu | Guangxiang Zhao | Zhuocheng Yu | Junfeng Ran | Xiangyu Wong | Lin Sun | Sujian Li
Findings of the Association for Computational Linguistics: ACL 2025
With the development of large language models (LLMs), there has been an increasing need for significant advancements in handling long contexts. To enhance long-context capabilities, constructing high-quality training data with long-range dependencies is crucial. Existing methods to select long-context data often rely on sentence-level analysis,which can be greatly optimized in both performance and efficiency. In this paper, we propose a novel token-level framework, LongAttn, which leverages the self-attention mechanism of LLMs to measure the long-range dependencies for the data. By calculating token-level dependency strength and distribution uniformity of token scores, LongAttn effectively quantifies long-range dependencies, enabling more accurate and efficient data selection. We filter LongABC-32K from open-source long-context datasets (ArXiv, Book, and Code). Through our comprehensive experiments, LongAttn has demonstrated its excellent effectiveness, scalability, and efficiency. We will release our code and the high-quality long-context dataset LongABC-32K in the future.
Hierarchical Memory Organization for Wikipedia Generation
Eugene J. Yu | Dawei Zhu | Yifan Song | Xiangyu Wong | Jiebin Zhang | Wenxuan Shi | Xiaoguang Li | Qun Liu | Sujian Li
Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Eugene J. Yu | Dawei Zhu | Yifan Song | Xiangyu Wong | Jiebin Zhang | Wenxuan Shi | Xiaoguang Li | Qun Liu | Sujian Li
Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Generating Wikipedia articles autonomously is a challenging task requiring the integration of accurate, comprehensive, and well-structured information from diverse sources. This paper introduces the Memory Organization-based Generation (MOG) framework, a novel approach to address these challenges by leveraging a hierarchical memory architecture. MOG extracts fine-grained memory units from web documents, recursively organizes them into a Wikipedia-style hierarchical structure, and uses this structure to guide the generation process. This ensures alignment between memory and the article outline, improving both informativeness and verifiability while minimizing hallucinations. Additionally, a citation module is implemented to enhance traceability by linking every generated sentence to specific memory units. Evaluations on our newly created WikiStart dataset demonstrate that MOG outperforms baseline methods in producing informative and reliable articles, making it particularly robust in real-world scenarios.
Search
Fix author
Co-authors
- Sujian Li (李素建) 6
- Xiaoguang Li 2
- Qun Liu 2
- Junfeng Ran 2
- Yifan Song 2
- Lin Sun 2
- Xiangyu Wong 2
- Wenhao Wu 2
- Jiebin Zhang 2
- Guangxiang Zhao 2
- Jiefeng Chen 1
- Hongxin Ding 1
- Chuqiao Kuang 1
- Zheng Li 1
- Rui Meng 1
- Tomas Pfister 1
- Lifeng Shang 1
- Wenxuan Shi 1
- Xiyu Wei 1
- Longyun Wu 1
- XWang 1
- Weimin Xiong 1
- Jinsung Yoon 1
- Eugene J. Yu 1
- Zhuocheng Yu 1
- Chenyu Zhang 1
- Xiangzheng Zhang 1
- Haosheng Zou 1