Yanbin Wei
Author directory2026
VizRAG: Enhancing Retrieval-Augmented Generation with Hypergraph Visualization
Yanbin Wei | Yang Chen | Renling Gan | Ziru Liu | Xinyu Fu | Chun Kang | Ning Lu | Rui Liu | Yu Zhang | James Kwok
Findings of the Association for Computational Linguistics: EMNLP 2026
Yanbin Wei | Yang Chen | Renling Gan | Ziru Liu | Xinyu Fu | Chun Kang | Ning Lu | Rui Liu | Yu Zhang | James Kwok
Findings of the Association for Computational Linguistics: EMNLP 2026
Hypergraph-based RAG systems surpass traditional graph-based approaches by organizing complex n-ary atomic facts among entities, rather than relying solely on binary relationships. Despite the advancements in multimodal large language models (MLLMs) with enhanced visual capabilities, current hypergraph-based RAG frameworks predominantly restrict knowledge retrieval and reconstruction to a unimodal, text-centric paradigm. This limitation prevents them from fully leveraging the powerful visual perception capabilities of modern MLLMs. To address this gap, we systematically explore the integration of hypergraph awareness in RAG systems through visual cues. By incorporating visual representations of hypergraphs into the RAG pipeline, we introduce VizRAG, the first RAG system to support visual hypergraph structure awareness. Experimental results demonstrate that VizRAG significantly outperforms strong baselines, validating the promising potential of hypergraph visualization as a novel approach for RAG systems.
HyperGVL: Benchmarking and Improving Large Vision-Language Models in Hypergraph Understanding and Reasoning
Yanbin Wei | Chun Kang | Siwei Li | Haoxuan Che | Yang Chen | Hua Liu | Jian Liu | Zhuang Liu | Can Ouyang | Fei Xing | Lei Sha | Rui Liu | Yu Zhang | James Kwok
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Yanbin Wei | Chun Kang | Siwei Li | Haoxuan Che | Yang Chen | Hua Liu | Jian Liu | Zhuang Liu | Can Ouyang | Fei Xing | Lei Sha | Rui Liu | Yu Zhang | James Kwok
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Large Vision-Language Models (LVLMs) consistently require new arenas to guide their expanding boundaries, yet their capabilities with hypergraphs remain unexplored. In the real world, hypergraphs have significant practical applications in areas such as life sciences and social communities. Recent advancements in LVLMs have shown promise in understanding complex topologies, yet there remains a lack of a benchmark to delineate the capabilities of LVLMs with hypergraphs, leaving the boundaries of their abilities unclear. To fill this gap, in this paper, we introduce HyperGVL, the first benchmark to evaluate the proficiency of LVLMs in hypergraph understanding and reasoning. HyperGVL provides a comprehensive assessment of 12 advanced LVLMs across 84,000 vision-language question-answering (QA) samples spanning 12 tasks, ranging from basic component counting to complex NP-hard problem reasoning. The involved hypergraphs contain multiscale synthetic structures and real-world citation and protein networks. Moreover, we examine the effects of 12 textual and visual hypergraph representations and introduce a generalizable router WiseHyGR that improves LVLMs in hypergraph via learning adaptive representations. We believe that this work is a step forward in connecting hypergraphs with LVLMs.
Policy and World Modeling Co-Training for Language Agents
Ning Lu | Baijiong Lin | Shengcai Liu | Jiahao Wu | Haoze Lv | Yanbin Wei | Lingting Zhu | Shengju Qian | Xin Wang | Ying-Cong Chen | Qi Wang | Ke Tang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Ning Lu | Baijiong Lin | Shengcai Liu | Jiahao Wu | Haoze Lv | Yanbin Wei | Lingting Zhu | Shengju Qian | Xin Wang | Ying-Cong Chen | Qi Wang | Ke Tang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Reinforcement learning (RL) improves large language model (LLM) agents by teaching them which actions lead to high rewards, but provides little supervision on what those actions do to the environment. World modeling (WM) can fill this gap, yet existing approaches often require separate simulators, extra training stages, or additional inference-time computation. We observe that on-policy RL rollouts already contain the needed signal: each transition pairs an action with its resulting next observation. Based on this observation, we propose PaW, a Policy and World modeling co-training framework that adds auxiliary WM supervision to the same policy during RL, without changing the inference paradigm. To make auxiliary WM supervision informative and stable, PaW introduces three components: action-entropy-based WM data selection, noise-tolerant WM loss, and reward-adaptive loss balancing. Experiments on three agentic task benchmarks show consistent improvements over strong RL baselines across models and RL algorithms. These results suggest that standard RL rollouts are a practical source of WM supervision for language-agent training.
TTT-Sparse: Test-Time Training of Sparse Attention for Long-Context Understanding
Xuehao Wang | Huayu Deng | Linhao Zhong | Yanbin Wei | Hengyu Zhao | Mingyu Liu | Hao Chen | Chunhua Shen
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Xuehao Wang | Huayu Deng | Linhao Zhong | Yanbin Wei | Hengyu Zhao | Mingyu Liu | Hao Chen | Chunhua Shen
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Pretrained LLMs excel across diverse tasks, yet standard attention remains a major bottleneck for long-context understanding. Recent test-time training (TTT) methods alleviate this by encoding context into internal model parameters, but they typically require specialized architectures, meta-learned initialization, or continual post-training, preventing plug-and-play use on off-the-shelf LLMs. We propose test-time training sparse attention (TTT-Sparse), a plug-and-play alternative that performs test-time training over external memory while keeping the internal parameters frozen. For each input, TTT-Sparse constructs contextual chunk key states from the KV cache and learns a lightweight query adapter to induce a task-specific sparse attention pattern, which is used during generation to attend only to the most relevant context chunks and suppress interference from irrelevant context. Extensive experiments on long-context benchmarks show that TTT-Sparse improves long-context understanding while reducing generation-time attention cost.
2023
KICGPT: Large Language Model with Knowledge in Context for Knowledge Graph Completion
Yanbin Wei | Qiushi Huang | Yu Zhang | James Kwok
Findings of the Association for Computational Linguistics: EMNLP 2023
Yanbin Wei | Qiushi Huang | Yu Zhang | James Kwok
Findings of the Association for Computational Linguistics: EMNLP 2023
Knowledge Graph Completion (KGC) is crucial for addressing knowledge graph incompleteness and supporting downstream applications. Many models have been proposed for KGC and they can be categorized into two main classes, including triple-based and test-based approaches. Triple-based methods struggle with long-tail entities due to limited structural information and imbalanced distributions of entities. Text-based methods alleviate this issue but require costly training for language models and specific finetuning for knowledge graphs, which limits their efficiency. To alleviate the limitations in the two approaches, in this paper, we propose KICGPT, a framework that integrates a large language model (LLM) and a triple-based KGC retriever, to alleviate the long-tail problem without incurring additional training overhead. In the proposed KICGPT model, we propose an in-context learning strategy called Knowledge Prompt, which encodes structural knowledge into demonstrations to guide LLM. Empirical results on benchmark datasets demonstrate the effectiveness of the proposed KICGPT model with lighter training overhead and no finetuning.
Search
Fix author
Co-authors
- James Kwok 3
- Yang Chen 2
- Chun Kang 2
- Rui Liu 2
- Ning Lu 2
- Yu Zhang 2
- Haoxuan Che 1
- Hao Chen 1
- Ying-Cong Chen 1
- Huayu Deng 1
- Xinyu Fu 1
- Renling Gan 1
- Qiushi Huang 1
- Siwei Li 1
- Baijiong Lin 1
- Hua Liu 1
- Jian Liu 1
- Mingyu Liu 1
- Shengcai Liu 1
- Zhuang Liu 1
- Ziru Liu 1
- Haoze Lv 1
- Can Ouyang 1
- Shengju Qian 1
- Lei Sha 1
- Chunhua Shen 1
- Ke Tang 1
- Qi Wang 1
- Xin Wang 1
- Xuehao Wang 1
- Jiahao Wu 1
- Fei Xing 1
- Yu Zhang 1
- Hengyu Zhao 1
- Linhao Zhong 1
- Lingting Zhu 1