Demo Data from SIGIR 2026 / reviewers in the wild / expert
Ningyu Zhang 0001
dblp:139/4181-1
· DBLP profile ↗
104ranked-venue papers
13as first author
95since 2021 · last 2026
0000-0002-1970-0678ORCID · conflict
Domains — the database's venue-derived domains; a paper can count in several
Artificial intelligence and machine learning · 89 · 11 first-author · 82 since 2021Graphics, computer vision, multimedia, augmented reality and games · 21 · 3 first-author · 20 since 2021Databases, data management, data science and information retrieval · 14 · 2 first-author · 11 since 2021Applied, interdisciplinary, general and emerging computing · 5 · 1 first-author · 3 since 2021Security and privacy · 1 · 1 first-author
Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.
| Artificial intelligence
72 papers |
Language models and text generation · 39% Information extraction and text analysis · 18% Knowledge representation and reasoning · 7% | |
| Databases, data mining, and information retrieval
17 papers |
Knowledge graphs · 93% Data mining · 7% |
Topics — the 30 heaviest of 134, each with the papers that count most for it
| Topic | Weight | Papers | Last | Evidence papers |
|---|---|---|---|---|
Natural language and speech › Language models and text generation
knowledge editing |
7.1 | 10 | 2025 | AnyEdit: Edit Any Knowledge Encoded in Language Models · ICML 2025 CaKE: Circuit-aware Editing Enables Generalizable Knowledge Learners · EMNLP 2025 Knowledge Circuits in Pretrained Transformers · NeurIPS 2024 |
Knowledge graphs
knowledge graph construction |
4.6 | 8 | 2024 | Continual Multimodal Knowledge Graph Construction · IJCAI 2024 Schema-aware Reference as Prompt Improves Data-Efficient Knowledge Graph Construction · SIGIR 2023 Construction and Applications of Billion-Scale Pre-Trained Multimodal Business Knowledge Graph · ICDE 2023 |
Natural language and speech › Information extraction and text analysis
relation extraction |
4.3 | 9 | 2023 | On Analyzing the Role of Image for Visual-Enhanced Relation Extraction (Student Abstract) · AAAI 2023 Ontology-enhanced Prompt-tuning for Few-shot Learning · WWW 2022 KnowPrompt: Knowledge-aware Prompt-tuning with Synergistic Optimization for Relation Extraction · WWW 2022 |
Natural language and speech › Language models and text generation
large language model |
2.7 | 4 | 2026 | Why Do Open-Source LLMs Struggle with Data Analysis? A Systematic Empirical Study · AAAI 2026 WISE: Rethinking the Knowledge Memory for Lifelong Model Editing of Large Language Models · NeurIPS 2024 SeqGPT: An Out-of-the-Box Large Language Model for Open Domain Sequence Understanding · AAAI 2024 |
Natural language and speech › Information extraction and text analysis › event extraction
event detection |
2.1 | 4 | 2023 | SPEECH: Structured Prediction with Energy-Based Event-Centric Hyperspheres · ACL (1) 2023 MLBiNet: A Cross-Sentence Collective Event Detection Network · ACL/IJCNLP (1) 2021 OntoED: Low-resource Event Detection with Ontology Embedding · ACL/IJCNLP (1) 2021 |
Knowledge graphs
link prediction |
2.0 | 4 | 2024 | Editing Language Model-Based Knowledge Graph Embeddings · AAAI 2024 NeuralKG: An Open Source Library for Diverse Representation Learning of Knowledge Graphs · SIGIR 2022 Relation Adversarial Network for Low Resource Knowledge Graph Completion · WWW 2020 |
Computer vision › Vision and language › vision-language model
multimodal large language model |
1.9 | 3 | 2025 | MLLM can see? Dynamic Correction Decoding for Hallucination Mitigation · ICLR 2025 Unified Hallucination Detection for Multimodal Large Language Models · ACL (1) 2024 ADS-Edit: A Multimodal Knowledge Editing Dataset for Autonomous Driving Systems · ACM Multimedia 2025 |
Natural language and speech › Language models and text generation
hallucination mitigation |
1.9 | 2 | 2026 | KnowRL: Exploring Knowledgeable Reinforcement Learning for Factuality · ACL (1) 2026 MLLM can see? Dynamic Correction Decoding for Hallucination Mitigation · ICLR 2025 |
Natural language and speech › Language models and text generation
model steering |
1.9 | 2 | 2026 | Why Steering Works: Toward a Unified View of Language Model Parameter Dynamics · ACL (1) 2026 Beyond Prompt Engineering: Robust Behavior Control in LLMs via Steering Target Atoms · ACL (1) 2025 |
Knowledge, reasoning and agents › Planning, search and constraint satisfaction › planning
agent planning |
1.8 | 2 | 2026 | Can We Predict Before Executing Machine Learning Agents? · ACL (1) 2026 Agent Planning with World Knowledge Model · NeurIPS 2024 |
Natural language and speech › Language models and text generation
hallucination detection |
1.5 | 2 | 2024 | FactCHD: Benchmarking Fact-Conflicting Hallucination Detection · IJCAI 2024 Unified Hallucination Detection for Multimodal Large Language Models · ACL (1) 2024 |
Machine learning › Trustworthy machine learning
robustness |
1.4 | 3 | 2026 | Beyond Prompt Engineering: Robust Behavior Control in LLMs via Steering Target Atoms · ACL (1) 2025 Illusions of Confidence? Diagnosing LLM Truthfulness via Neighborhood Consistency · ACL (1) 2026 FactCHD: Benchmarking Fact-Conflicting Hallucination Detection · IJCAI 2024 |
Natural language and speech › Information extraction and text analysis
event extraction |
1.4 | 3 | 2024 | SPEECH: Structured Prediction with Energy-Based Event-Centric Hyperspheres · ACL (1) 2023 Contrastive Information Extraction With Generative Transformer · IEEE ACM Trans. Audio Speech Lang. Process. 2021 SeqGPT: An Out-of-the-Box Large Language Model for Open Domain Sequence Understanding · AAAI 2024 |
Computer vision › Vision and language › vision-language model
prompt learning |
1.3 | 3 | 2023 | Decoupling Knowledge from Memorization: Retrieval-augmented Prompt Learning · NeurIPS 2022 Differentiable Prompt Makes Pre-trained Language Models Better Few-shot Learners · ICLR 2022 Knowledge Rumination for Pre-trained Language Models · EMNLP 2023 |
Knowledge graphs
knowledge graph embedding |
1.3 | 2 | 2024 | Editing Language Model-Based Knowledge Graph Embeddings · AAAI 2024 NeuralKG: An Open Source Library for Diverse Representation Learning of Knowledge Graphs · SIGIR 2022 |
Natural language and speech › Information extraction and text analysis › relation extraction
triple extraction |
1.2 | 3 | 2023 | Contrastive Information Extraction With Generative Transformer · IEEE ACM Trans. Audio Speech Lang. Process. 2021 Contrastive Triple Extraction with Generative Transformer · AAAI 2021 Schema-aware Reference as Prompt Improves Data-Efficient Knowledge Graph Construction · SIGIR 2023 |
Natural language and speech › Language models and text generation
prompt tuning |
1.1 | 2 | 2022 | Ontology-enhanced Prompt-tuning for Few-shot Learning · WWW 2022 KnowPrompt: Knowledge-aware Prompt-tuning with Synergistic Optimization for Relation Extraction · WWW 2022 |
Natural language and speech › Language models and text generation › large language model › knowledge in language models
knowledge injection |
1.1 | 2 | 2022 | Ontology-enhanced Prompt-tuning for Few-shot Learning · WWW 2022 Drop Redundant, Shrink Irrelevant: Selective Knowledge Injection for Language Pretraining · IJCAI 2021 |
Knowledge, reasoning and agents › Knowledge representation and reasoning › ontology
ontology embedding |
1.1 | 2 | 2022 | OntoProtein: Protein Pretraining With Gene Ontology Embedding · ICLR 2022 OntoED: Low-resource Event Detection with Ontology Embedding · ACL/IJCNLP (1) 2021 |
Robotics › Motion planning and robot control
controllability |
1.0 | 1 | 2026 | How Controllable Are Large Language Models? A Unified Evaluation across Behavioral Granularities · ACL (1) 2026 |
Machine learning › Reinforcement learning
knowledge-based reinforcement learning |
1.0 | 1 | 2026 | KnowRL: Exploring Knowledgeable Reinforcement Learning for Factuality · ACL (1) 2026 |
Machine learning › Representation and self-supervised learning
neighborhood consistency |
1.0 | 1 | 2026 | Illusions of Confidence? Diagnosing LLM Truthfulness via Neighborhood Consistency · ACL (1) 2026 |
Natural language and speech › Language models and text generation › large language model evaluation
truthfulness evaluation |
1.0 | 1 | 2026 | Illusions of Confidence? Diagnosing LLM Truthfulness via Neighborhood Consistency · ACL (1) 2026 |
Program synthesis and code generation
code generation with language models |
1.0 | 1 | 2026 | ReCode: Updating Code API Knowledge with Reinforcement Learning · AAAI 2026 |
Machine learning › Trustworthy machine learning
interpretability |
1.0 | 2 | 2024 | Knowledge Circuits in Pretrained Transformers · NeurIPS 2024 Tool-Augmented Reward Modeling · ICLR 2024 |
Robotics › Motion planning and robot control › robot control
behavior control |
0.9 | 1 | 2025 | Beyond Prompt Engineering: Robust Behavior Control in LLMs via Steering Target Atoms · ACL (1) 2025 |
Natural language and speech › Language models and text generation › large language model inference
inference-time steering |
0.9 | 1 | 2025 | Two Experts Are All You Need for Steering Thinking: Reinforcing Cognitive Effort in MoE Reasoning Models Without Additional Training · NeurIPS 2025 |
Knowledge, reasoning and agents › Knowledge representation and reasoning › belief change
knowledge forgetting |
0.9 | 1 | 2025 | ReLearn: Unlearning via Learning for Large Language Models · ACL (1) 2025 |
Natural language and speech › Language models and text generation
large language model evaluation |
0.9 | 1 | 2025 | ReLearn: Unlearning via Learning for Large Language Models · ACL (1) 2025 |
Natural language and speech › Language models and text generation › large language model
large reasoning model |
0.9 | 1 | 2025 | Two Experts Are All You Need for Steering Thinking: Reinforcing Cognitive Effort in MoE Reasoning Models Without Additional Training · NeurIPS 2025 |
Methods — techniques the papers use, named apart from their topics
instruction tuning · 4.5large language model · 3.0multimodal learning · 2.8contrastive learning · 2.5knowledge editing · 2.4activation steering · 1.9data synthesis · 1.8prompt tuning · 1.7circuit analysis · 1.6benchmark construction · 1.5uniprot · 1.3gene ontology · 1.3retrieval augmentation · 1.2prompt learning · 1.2transformer · 1.2string similarity metric · 1.0reinforcement learning · 1.0GRPO · 1.0
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2026 | ReCode: Updating Code API Knowledge with Reinforcement LearningabstractLarge Language Models (LLMs) exhibit remarkable code generation capabilities but falter when adapting to frequent updates in external library APIs. This critical limitation, stemming from reliance on outdated API knowledge from their training data, even with access to current documentation, impedes reliable code generation in dynamic environments. To tackle this issue, we propose ReCode (rule-based Reinforcement learning for Code Update), a novel framework that mimics human programmer adaptation to API changes. Specifically, we construct a dataset of approximately 2,000 data entries to train the LLMs to perform version migration based on updated information. Then, we introduce a modified string similarity metric for code evaluation as the reward for reinforcement learning. Our experiments demonstrate that ReCode substantially boosts LLMs' code generation performance in dynamic API scenarios, especially on the unseen CodeUpdateArena task. Crucially, compared to supervised fine-tuning, ReCode has less impact on LLMs' general code generation abilities. We apply ReCode on various LLMs and reinforcement learning algorithms (GRPO and DAPO), all achieving consistent improvements. Notably, after training, Qwen2.5-Coder-7B outperforms that of the 32B parameter code instruction-tuned model and the reasoning model with the same architecture. Yunzhi Yao, Wenhao Yu 0002, Ningyu Zhang 0001 |
AAAI | 4 |
| 2026 | Why Do Open-Source LLMs Struggle with Data Analysis? A Systematic Empirical StudyabstractLarge Language Models (LLMs) hold promise in automating data analysis tasks, yet open-source models face significant limitations in these kinds of reasoning-intensive scenarios. In this work, we investigate strategies to enhance the data analysis capabilities of open-source LLMs. By curating a seed dataset of diverse, realistic scenarios, we evaluate models across three dimensions: data understanding, code generation, and strategic planning. Our analysis reveals three key findings: (1) Strategic planning quality serves as the primary determinant of model performance; (2) Interaction design and task complexity significantly influence reasoning capabilities; (3) Data quality demonstrates a greater impact than diversity in achieving optimal performance. We leverage these insights to develop a data synthesis methodology, demonstrating significant improvements in open-source LLMs' analytical reasoning capabilities. Jintian Zhang, Shuofei Qiao, Lun Du, Da Zheng 0004, Ningyu Zhang 0001, Huajun Chen |
AAAI | 9 |
| 2026 | KnowRL: Exploring Knowledgeable Reinforcement Learning for FactualityabstractSlow-thinking Large Language Models (LLMs) have demonstrated strong reasoning capabilities but often suffer from severe hallucinations due to an inability to recognize their knowledge boundaries.Existing Reinforcement Learning (RL) approaches typically rely on outcomeoriented rewards, which can inadvertently reinforce fabricated reasoning paths when the final answer is correct.To address this, we propose Knowledge-enhanced RL, KnowRL, a framework that integrates factual supervision directly into the reasoning process.By decomposing the chain of thought into atomic facts and verifying them against the corresponding ground-truth knowledge, KnowRL performs fine-grained checks to encourage models to reason faithfully.Crucially, this processoriented supervision teaches the model to identify its knowledge boundaries, learning to say "I don't know" instead of fabricating answers when information is missing.Experimental results demonstrate that KnowRL effectively mitigates hallucinations-reducing the Incorrect Rate on SimpleQA by 20.3% for distillationbased slow-thinking models while maintaining strong performance on complex reasoning benchmarks like GPQA and AIME 2025.Furthermore, our method shows robust transferability to out-of-distribution tasks, indicating that the model learns a generalizable verification behavior 1 . Baochang Ren, Shuofei Qiao, Ningyu Zhang 0001, Da Zheng 0004, Huajun Chen |
ACL (1) | 3 |
| 2026 | Why Steering Works: Toward a Unified View of Language Model Parameter DynamicsabstractZiwen Xu, Chenyan WU, Hengyu Sun, Haiwen Hong, Mengru Wang, Yunzhi Yao, Longtao Huang, Hui Xue, Shumin Deng, Zhixuan Chu, Huajun Chen, Ningyu Zhang. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Ziwen Xu, Chenyan Wu, Hengyu Sun, Haiwen Hong, Yunzhi Yao, Longtao Huang, Hui Xue 0001, Shumin Deng, Zhixuan Chu, Huajun Chen, Ningyu Zhang 0001 |
ACL (1) | 12 |
| 2026 | How Controllable Are Large Language Models? A Unified Evaluation across Behavioral GranularitiesabstractZiwen Xu, Kewei Xu, Haoming Xu, Haiwen Hong, Longtao Huang, Hui Xue, Ningyu Zhang, Yongliang Shen, Guozhou Zheng, Huajun Chen, Shumin Deng. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Ziwen Xu, Kewei Xu, Haiwen Hong, Longtao Huang, Hui Xue 0001, Ningyu Zhang 0001, Yongliang Shen 0001, Guozhou Zheng, Huajun Chen, Shumin Deng |
ACL (1) | 7 |
| 2026 | Illusions of Confidence? Diagnosing LLM Truthfulness via Neighborhood ConsistencyabstractHaoming Xu, Ningyuan Zhao, Yunzhi Yao, Weihong Xu, Hongru Wang, Xinle Deng, Shumin Deng, Jeff Z. Pan, Huajun Chen, Ningyu Zhang. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Ningyuan Zhao, Yunzhi Yao, Hongru Wang 0003, Xinle Deng, Shumin Deng, Jeff Z. Pan, Huajun Chen, Ningyu Zhang 0001 |
ACL (1) | 10 |
| 2026 | Can We Predict Before Executing Machine Learning Agents?abstractJingsheng Zheng, Jintian Zhang, Yujie Luo, Yuren Mao, Yunjun Gao, Lun Du, Huajun Chen, Ningyu Zhang. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Jingsheng Zheng, Jintian Zhang, Yuren Mao, Yunjun Gao, Lun Du, Huajun Chen, Ningyu Zhang 0001 |
ACL (1) | 8 |
| 2025 | CKnowEdit: A New Chinese Knowledge Editing Dataset for Linguistics, Facts, and Logic Error Correction in LLMsabstractJizhan Fang, Tianhe Lu, Yunzhi Yao, Ziyan Jiang, Xin Xu, Huajun Chen, Ningyu Zhang. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025. Jizhan Fang, Tianhe Lu, Yunzhi Yao, Ziyan Jiang, Xin Xu 0010, Huajun Chen, Ningyu Zhang 0001 |
ACL (1) | 7 |
| 2025 | Agentic Knowledgeable Self-awarenessabstractShuofei Qiao, Zhisong Qiu, Baochang Ren, Xiaobin Wang, Xiangyuan Ru, Ningyu Zhang, Xiang Chen, Yong Jiang, Pengjun Xie, Fei Huang, Huajun Chen. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025. Shuofei Qiao, Zhisong Qiu, Baochang Ren, Xiaobin Wang, Xiangyuan Ru, Ningyu Zhang 0001, Xiang Chen 0016, Yong Jiang 0005, Pengjun Xie, Fei Huang 0002, Huajun Chen |
ACL (1) | 6 |
| 2025 | Beyond Prompt Engineering: Robust Behavior Control in LLMs via Steering Target AtomsabstractMengru Wang, Ziwen Xu, Shengyu Mao, Shumin Deng, Zhaopeng Tu, Huajun Chen, Ningyu Zhang. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025. Ziwen Xu, Shengyu Mao, Shumin Deng, Zhaopeng Tu, Huajun Chen, Ningyu Zhang 0001 |
ACL (1) | 7 |
| 2025 | ReLearn: Unlearning via Learning for Large Language ModelsabstractCurrent unlearning methods for large language models usually rely on reverse optimization to reduce target token probabilities. However, this paradigm disrupts the subsequent tokens prediction, degrading model performance and linguistic coherence. Moreover, existing evaluation metrics overemphasize contextual forgetting while inadequately assessing response fluency and relevance. To address these challenges, we propose ReLearn, a data augmentation and fine-tuning pipeline for effective unlearning, along with a comprehensive evaluation framework. This framework introduces Knowledge Forgetting Ratio (KFR) and Knowledge Retention Ratio (KRR) to measure knowledge-level preservation, and Linguistic Score (LS) to evaluate generation quality. Our experiments show that ReLearn successfully achieves targeted forgetting while preserving high-quality outputs. Through mechanistic analysis, we further demonstrate how reverse optimization disrupts coherent text generation, while ReLearn preserves this essential capability. Ningyuan Zhao, Sendong Zhao, Shumin Deng, Bryan Hooi, Nay Oo, Huajun Chen, Ningyu Zhang 0001 |
ACL (1) | 10 |
| 2025 | Knowledge Graph Pooling and Unpooling for Concept AbstractionabstractKnowledge graph embedding (KGE) aims to embed entities and relations as vectors in a continuous space and has proven to be effective for KG tasks. Recently, graph neural networks (GNN) based KGEs gain much attention due to their strong capability of encoding complex graph structures. However, most GNN-based KGEs are directly optimized based on the instance triples in KGs, ignoring the latent concepts and hierarchies of the entities. Though some works explicitly inject concepts and hierarchies into models, they are limited to predefined concepts and hierarchies, which are missing in a lot of KGs. Thus in this paper, we propose a novel framework with KG Pooling and unpooling and Contrastive Learning (KGPCL) to abstract and encode the latent concepts for better KG prediction. Specifically, with an input KG, we first construct a U-KG through KG pooling and unpooling. KG pooling abstracts the input graph to a smaller graph as a pooled graph, and KG unpooling recovers the input graph from the pooled graph. Then we model the U-KG with relational KGEs to get the representations of entities and relations for prediction. Finally, we propose the local and global contrastive loss to jointly enhance the representation of entities. Experimental results show that our models outperform the KGE baselines on link prediction task. Juan Li 0010, Wen Zhang 0015, Mingchen Tu, Mingyang Chen 0002, Ningyu Zhang 0001, Shijian Li |
COLING | 6 |
| 2025 | OmniThink: Expanding Knowledge Boundaries in Machine Writing through ThinkingabstractZekun Xi, Wenbiao Yin, Jizhan Fang, Jialong Wu, Runnan Fang, Yong Jiang, Pengjun Xie, Fei Huang, Huajun Chen, Ningyu Zhang. Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing. 2025. Zekun Xi, Wenbiao Yin, Jizhan Fang, Jialong Wu 0007, Runnan Fang, Yong Jiang 0005, Pengjun Xie, Fei Huang 0002, Huajun Chen, Ningyu Zhang 0001 |
EMNLP | 10 |
| 2025 | CaKE: Circuit-aware Editing Enables Generalizable Knowledge LearnersabstractKnowledge Editing (KE) enables the modification of outdated or incorrect information in large language models (LLMs).While existing KE methods can update isolated facts, they often fail to generalize these updates to multihop reasoning tasks that rely on the modified knowledge.Through an analysis of reasoning circuits-the neural pathways LLMs use for knowledge-based inference, we find that current layer-localized KE approaches (e.g., MEMIT, WISE), which edit only single or a few model layers, inadequately integrate updated knowledge into these reasoning pathways.To address this limitation, we present CaKE (Circuit-aware Knowledge Editing), a novel method that enhances the effective integration of updated knowledge in LLMs.By only leveraging a few curated data samples guided by our circuit-based analysis, CaKE stimulates the model to develop appropriate reasoning circuits for newly incorporated knowledge.Experiments show that CaKE enables more accurate and consistent use of edited knowledge across related reasoning tasks, achieving an average improvement of 20% in multi-hop reasoning accuracy on the MQuAKE dataset while requiring less memory than existing KE methods.We release the code and data in https://github.com/zjunlp/CaKE. Yunzhi Yao, Jizhan Fang, Jia-Chen Gu, Ningyu Zhang 0001, Shumin Deng, Huajun Chen, Nanyun Peng 0001 |
EMNLP | 4 |
| 2025 | LightThinker: Thinking Step-by-Step CompressionabstractJintian Zhang, Yuqi Zhu, Mengshu Sun, Yujie Luo, Shuofei Qiao, Lun Du, Da Zheng, Huajun Chen, Ningyu Zhang. Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing. 2025. Jintian Zhang, Mengshu Sun, Shuofei Qiao, Lun Du, Da Zheng 0004, Huajun Chen, Ningyu Zhang 0001 |
EMNLP | 9 |
| 2025 | Benchmarking Agentic Workflow GenerationabstractLarge Language Models (LLMs), with their exceptional ability to handle a wide range of tasks, have driven significant advancements in tackling reasoning and planning tasks, wherein decomposing complex problems into executable workflows is a crucial step in this process. Existing workflow evaluation frameworks either focus solely on holistic performance or suffer from limitations such as restricted scenario coverage, simplistic workflow structures, and lax evaluation standards. To this end, we introduce WorfBench, a unified workflow generation benchmark with multi-faceted scenarios and intricate graph workflow structures. Additionally, we present WorfEval, a systemic evaluation protocol utilizing subsequence and subgraph matching algorithms to accurately quantify the LLM agent's workflow generation capabilities. Through comprehensive evaluations across different types of LLMs, we discover distinct gaps between the sequence planning capabilities and graph planning capabilities of LLM agents, with even GPT-4 exhibiting a gap of around 15%. We also train two open-source models and evaluate their generalization abilities on held-out tasks. Furthermore, we observe that the generated workflows can enhance downstream tasks, enabling them to achieve superior performance with less time during inference. Code and dataset are available at https://github.com/zjunlp/WorfBench. Shuofei Qiao, Runnan Fang, Zhisong Qiu, Xiaobin Wang, Ningyu Zhang 0001, Yong Jiang 0005, Pengjun Xie, Fei Huang 0002, Huajun Chen |
ICLR | 5 |
| 2025 | MLLM can see? Dynamic Correction Decoding for Hallucination MitigationabstractMultimodal Large Language Models (MLLMs) frequently exhibit hallucination phenomena, but the underlying reasons remain poorly understood. In this paper, we present an empirical analysis and find that, although MLLMs incorrectly generate the objects in the final output, they are actually able to recognize visual objects in the preceding layers. We speculate that this may be due to the strong knowledge priors of the language model suppressing the visual information, leading to hallucinations. Motivated by this, we propose a novel dynamic correction decoding method for MLLMs DeCo, which adaptively selects the appropriate preceding layers and proportionally integrates knowledge into the final layer to adjust the output logits. Note that DeCo is model agnostic and can be seamlessly incorporated with various classic decoding strategies and applied to different MLLMs. We evaluate DeCo on widely-used benchmarks, demonstrating that it can reduce hallucination rates by a large margin compared to baselines, highlighting its potential to mitigate hallucinations. Code is available at https://github.com/zjunlp/DeCo. Xiang Chen 0016, Ningyu Zhang 0001, Bozhong Tian, Shumin Deng, Huajun Chen |
ICLR | 3 |
| 2025 | AnyEdit: Edit Any Knowledge Encoded in Language ModelsabstractLarge language models (LLMs) often produce incorrect or outdated information, necessitating efficient and precise knowledge updates. Current model editing methods, however, struggle with long-form knowledge in diverse formats, such as poetry, code snippets, and mathematical derivations. These limitations arise from their reliance on editing a single token’s hidden state, a limitation we term as ``efficacy barrier''. To solve this, we propose \textbf{AnyEdit}, a new autoregressive editing paradigm. It decomposes long-form knowledge into sequential chunks and iteratively edits the key token in each chunk, ensuring consistent and accurate outputs. Theoretically, we ground AnyEdit in the Chain Rule of Mutual Information, showing its ability to update any knowledge within LLMs. Empirically, it outperforms strong baselines by 21.5\% on benchmarks including UnKEBench, AKEW, and our new \textbf{EditEverything} dataset for long-form diverse-formatted knowledge. Additionally, AnyEdit serves as a plug-and-play framework, enabling current editing methods to update knowledge with arbitrary length and format, significantly advancing the scope and practicality of LLM knowledge editing. Our code is available at: \url{https://github.com/jianghoucheng/AnyEdit}. Houcheng Jiang, Junfeng Fang, Ningyu Zhang 0001, Mingyang Wan, Guojun Ma, Xiang Wang 0010, Xiangnan He 0001, Tat-Seng Chua |
ICML | 3 |
| 2025 | ADS-Edit: A Multimodal Knowledge Editing Dataset for Autonomous Driving Systems
Jizhan Fang, Xiang Chen 0016, Bozhong Tian, Ziwen Xu, Huajun Chen, Ningyu Zhang 0001 |
ACM Multimedia | 7 |
| 2025 | RSCC: A Large-Scale Remote Sensing Change Caption Dataset for Disaster EventsabstractRemote sensing is critical for disaster monitoring, yet existing datasets lack temporal image pairs and detailed textual annotations. While single-snapshot imagery dominates current resources, it fails to capture dynamic disaster impacts over time. To address this gap, we introduce the Remote Sensing Change Caption (RSCC) dataset, a large-scale benchmark comprising 62,351 pre-/post-disaster image pairs (spanning earthquakes, floods, wildfires, and more) paired with rich, human-like change captions. By bridging the temporal and semantic divide in remote sensing data, RSCC enables robust training and evaluation of vision-language models for disaster-aware bi-temporal understanding. Our results highlight RSCC’s ability to facilitate detailed disaster-related analysis, paving the way for more accurate, interpretable, and scalable vision-language applications in remote sensing. Code and dataset are available at https://github.com/Bili-Sakura/RSCC. Zhenyuan Chen, Ningyu Zhang 0001 |
NeurIPS | 3 |
| 2025 | Two Experts Are All You Need for Steering Thinking: Reinforcing Cognitive Effort in MoE Reasoning Models Without Additional TrainingabstractMixture-of-Experts (MoE) architectures within Large Reasoning Models (LRMs) have achieved impressive reasoning capabilities by selectively activating experts to facilitate structured cognitive processes. Despite notable advances, existing reasoning models often suffer from cognitive inefficiencies like overthinking and underthinking. To address these limitations, we introduce a novel inference-time steering methodology called Reinforcing Cognitive Experts (RICE), designed to improve reasoning depth and efficiency without additional training or complex heuristics. Leveraging normalized Pointwise Mutual Information (nPMI), we systematically identify specialized experts, termed cognitive experts that orchestrate meta-level reasoning operations characterized by tokens like <think>. Empirical evaluations with leading MoE-based LRMs (DeepSeek-R1 and Qwen3-235B) on rigorous quantitative and scientific reasoning benchmarks (AIME and GPQA Diamond) demonstrate noticeable and consistent improvements in reasoning accuracy, cognitive efficiency, and cross-domain generalization. Crucially, our lightweight approach substantially outperforms prevalent reasoning-steering techniques, such as prompt design and decoding constraints, while preserving the model's general instruction-following skills. These results highlight reinforcing cognitive experts as a promising, practical, and interpretable direction to enhance cognitive efficiency within advanced reasoning models. Yue Wang 0039, Zhiwei He 0002, Qiuzhi Liu, Yunzhi Yao, Wenxuan Wang 0001, Ruotian Ma, Haitao Mi, Ningyu Zhang 0001, Zhaopeng Tu, Dong Yu 0001 |
NeurIPS | 12 |
| 2024 | Editing Language Model-Based Knowledge Graph EmbeddingsabstractRecently decades have witnessed the empirical success of framing Knowledge Graph (KG) embeddings via language models. However, language model-based KG embeddings are usually deployed as static artifacts, making them difficult to modify post-deployment without re-training after deployment. To address this issue, we propose a new task of editing language model-based KG embeddings in this paper. This task is designed to facilitate rapid, data-efficient updates to KG embeddings without compromising the performance of other aspects. We build four new datasets: E-FB15k237, A-FB15k237, E-WN18RR, and A-WN18RR, and evaluate several knowledge editing baselines demonstrating the limited ability of previous models to handle the proposed challenging task. We further propose a simple yet strong baseline dubbed KGEditor, which utilizes additional parametric layers of the hypernetwork to edit/add facts. Our comprehensive experimental results reveal that KGEditor excels in updating specific facts without impacting the overall performance, even when faced with limited training resources. Code and datasets will be available at https://github.com/AnonymousForPapers/DeltaKG. Siyuan Cheng 0008, Ningyu Zhang 0001, Bozhong Tian, Xi Chen 0003, Qingbin Liu, Huajun Chen |
AAAI | 2 |
| 2024 | When Do Program-of-Thought Works for Reasoning?abstractIn the realm of embodied artificial intelligence, the reasoning capabilities of Large Language Models (LLMs) play a pivotal role. Although there are effective methods like program-of-thought prompting for LLMs which uses programming language to tackle complex reasoning tasks, the specific impact of code data on the improvement of reasoning capabilities remains under-explored. To address this gap, we propose complexity-impacted reasoning score CIRS, which combines structural and logical attributes, to measure the correlation between code and reasoning abilities. Specifically, we use the abstract syntax tree to encode the structural information and calculate logical complexity by considering the difficulty and the cyclomatic complexity. Through an empirical analysis, we find not all code data of complexity can be learned or understood by LLMs. Optimal level of complexity is critical to the improvement of reasoning abilities by program-aided prompting. Then we design an auto-synthesizing and stratifying algorithm, and apply it to instruction generation for mathematical reasoning and code data filtering for code generation tasks. Extensive results demonstrates the effectiveness of our proposed approach. Zhen Bi, Ningyu Zhang 0001, Yinuo Jiang, Shumin Deng, Guozhou Zheng, Huajun Chen |
AAAI | 2 |
| 2024 | Existence Is Chaos: Enhancing 3D Human Motion Prediction with Uncertainty ConsiderationabstractHuman motion prediction is consisting in forecasting future body poses from historically observed sequences. It is a longstanding challenge due to motion's complex dynamics and uncertainty. Existing methods focus on building up complicated neural networks to model the motion dynamics. The predicted results are required to be strictly similar to the training samples with L2 loss in current training pipeline. However, little attention has been paid to the uncertainty property which is crucial to the prediction task. We argue that the recorded motion in training data could be an observation of possible future, rather than a predetermined result. In addition, existing works calculate the predicted error on each future frame equally during training, while recent work indicated that different frames could play different roles. In this work, a novel computationally efficient encoder-decoder model with uncertainty consideration is proposed, which could learn proper characteristics for future frames by a dynamic function. Experimental results on benchmark datasets demonstrate that our uncertainty consideration approach has obvious advantages both in quantity and quality. Moreover, the proposed method could produce motion sequences with much better quality that avoids the intractable shaking artefacts. We believe our work could provide a novel perspective to consider the uncertainty quality for the general motion prediction task and encourage the studies in this field. The code will be available in https://github.com/Motionpre/Adaptive-Salient-Loss-SAGGB. Ningyu Zhang 0001, Xiaosong Yang, Jun Xiao 0001 |
AAAI | 3 |
| 2024 | SeqGPT: An Out-of-the-Box Large Language Model for Open Domain Sequence UnderstandingabstractLarge language models (LLMs) have shown impressive abilities for open-domain NLP tasks. However, LLMs are sometimes too footloose for natural language understanding (NLU) tasks which always have restricted output and input format. Their performances on NLU tasks are highly related to prompts or demonstrations and are shown to be poor at performing several representative NLU tasks, such as event extraction and entity typing. To this end, we present SeqGPT, a bilingual (i.e., English and Chinese) open-source autoregressive model specially enhanced for open-domain natural language understanding. We express all NLU tasks with two atomic tasks, which define fixed instructions to restrict the input and output format but still ``open'' for arbitrarily varied label sets. The model is first instruction-tuned with extremely fine-grained labeled data synthesized by ChatGPT and then further fine-tuned by 233 different atomic tasks from 152 datasets across various domains. The experimental results show that SeqGPT has decent classification and extraction ability, and is capable of performing language understanding tasks on unseen domains. We also conduct empirical studies on the scaling of data and model size as well as on the transfer across tasks. Our models are accessible at https://github.com/Alibaba-NLP/SeqGPT. Tianyu Yu 0002, Chengyue Jiang, Chao Lou, Shen Huang, Xiaobin Wang, Wei Liu 0131, Jiong Cai, Yangning Li, Kewei Tu, Hai-Tao Zheng 0002, Ningyu Zhang 0001, Pengjun Xie, Fei Huang 0002, Yong Jiang 0005 |
AAAI | 12 |
| 2024 | OceanGPT: A Large Language Model for Ocean Science TasksabstractZhen Bi, Ningyu Zhang, Yida Xue, Yixin Ou, Daxiong Ji, Guozhou Zheng, Huajun Chen. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024. Zhen Bi, Ningyu Zhang 0001, Yida Xue, Yixin Ou, Daxiong Ji, Guozhou Zheng, Huajun Chen |
ACL (1) | 2 |
| 2024 | Unified Hallucination Detection for Multimodal Large Language ModelsabstractXiang Chen, Chenxi Wang, Yida Xue, Ningyu Zhang, Xiaoyan Yang, Qiang Li, Yue Shen, Lei Liang, Jinjie Gu, Huajun Chen. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024. Xiang Chen 0016, Yida Xue, Ningyu Zhang 0001, Qiang Li 0022, Lei Liang 0002, Jinjie Gu, Huajun Chen |
ACL (1) | 4 |
| 2024 | AutoAct: Automatic Agent Learning from Scratch for QA via Self-PlanningabstractShuofei Qiao, Ningyu Zhang, Runnan Fang, Yujie Luo, Wangchunshu Zhou, Yuchen Jiang, Chengfei Lv, Huajun Chen. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024. Shuofei Qiao, Ningyu Zhang 0001, Runnan Fang, Wangchunshu Zhou, Yuchen Eleanor Jiang, Chengfei Lv, Huajun Chen |
ACL (1) | 2 |
| 2024 | Detoxifying Large Language Models via Knowledge EditingabstractMengru Wang, Ningyu Zhang, Ziwen Xu, Zekun Xi, Shumin Deng, Yunzhi Yao, Qishen Zhang, Linyi Yang, Jindong Wang, Huajun Chen. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024. Ningyu Zhang 0001, Ziwen Xu, Zekun Xi, Shumin Deng, Yunzhi Yao, Qishen Zhang, Linyi Yang, Jindong Wang 0001, Huajun Chen |
ACL (1) | 2 |
| 2024 | Exploring Collaboration Mechanisms for LLM Agents: A Social Psychology ViewabstractAs Natural Language Processing (NLP) systems are increasingly employed in intricate social environments, a pressing query emerges: Can these NLP systems mirror human-esque collaborative intelligence, in a multi-agent society consisting of multiple large language models (LLMs)? This paper probes the collaboration mechanisms among contemporary NLP systems by melding practical experiments with theoretical insights. We fabricate four unique ‘societies’ comprised of LLM agents, where each agent is characterized by a specific ‘trait’ (easy-going or overconfident) and engages in collaboration with a distinct ‘thinking pattern’ (debate or reflection). Through evaluating these multi-agent societies on three benchmark datasets, we discern that certain collaborative strategies not only outshine previous top-tier approaches but also optimize efficiency (using fewer API tokens). Moreover, our results further illustrate that LLM agents manifest human-like social behaviors, such as conformity and consensus reaching, mirroring foundational social psychology theories. In conclusion, we integrate insights from social psychology to contextualize the collaboration of LLM agents, inspiring further investigations into the collaboration mechanism for LLMs. We commit to sharing our code and datasets, hoping to catalyze further research in this promising avenue. Jintian Zhang, Xin Xu 0010, Ningyu Zhang 0001, Ruibo Liu, Bryan Hooi, Shumin Deng |
ACL (1) | 3 |
| 2024 | Tool-Augmented Reward ModelingabstractReward modeling (*a.k.a.*, preference modeling) is instrumental for aligning large language models with human preferences, particularly within the context of reinforcement learning from human feedback (RLHF). While conventional reward models (RMs) have exhibited remarkable scalability, they oft struggle with fundamental functionality such as arithmetic computation, code execution, and factual lookup. In this paper, we propose a tool-augmented preference modeling approach, named Themis, to address these limitations by empowering RMs with access to external environments, including calculators and search engines. This approach not only fosters synergy between tool utilization and reward grading but also enhances interpretive capacity and scoring reliability. Our study delves into the integration of external tools into RMs, enabling them to interact with diverse external sources and construct task-specific tool engagement and reasoning traces in an autoregressive manner. We validate our approach across a wide range of domains, incorporating seven distinct external tools. Our experimental results demonstrate a noteworthy overall improvement of 17.7% across eight tasks in preference ranking. Furthermore, our approach outperforms Gopher 280B by 7.3% on TruthfulQA task in zero-shot evaluation. In human evaluations, RLHF trained with Themis attains an average win rate of 32% when compared to baselines across four distinct tasks. Additionally, we provide a comprehensive collection of tool-related RM datasets, incorporating data from seven distinct tool APIs, totaling 15,000 instances. We have made the code, data, and model checkpoints publicly available to facilitate and inspire further research advancements (https://github.com/ernie-research/Tool-Augmented-Reward-Model). Lei Li 0040, Yekun Chai, Shuohuan Wang, Yu Sun 0004, Hao Tian 0005, Ningyu Zhang 0001, Hua Wu 0003 |
ICLR | 6 |
| 2024 | Domain-Agnostic Molecular Generation with Chemical FeedbackabstractThe generation of molecules with desired properties has become increasingly popular, revolutionizing the way scientists design molecular structures and providing valuable support for chemical and drug design. However, despite the potential of language models in molecule generation, they face challenges such as generating syntactically or chemically flawed molecules, having narrow domain focus, and struggling to create diverse and feasible molecules due to limited annotated data or external molecular databases.
To tackle these challenges, we introduce MolGen, a pre-trained molecular language model tailored specifically for molecule generation. Through the reconstruction of over 100 million molecular SELFIES, MolGen internalizes structural and grammatical insights. This is further enhanced by domain-agnostic molecular prefix tuning, fostering robust knowledge transfer across diverse domains. Importantly, our chemical feedback paradigm steers the model away from "molecular hallucinations", ensuring alignment between the model's estimated probabilities and real-world chemical preferences. Extensive experiments on well-known benchmarks underscore MolGen's optimization capabilities in properties such as penalized logP, QED, and molecular docking. Additional analyses confirm its proficiency in accurately capturing molecule distributions, discerning intricate structural patterns, and efficiently exploring the chemical space (https://github.com/zjunlp/MolGen). Yin Fang, Ningyu Zhang 0001, Zhuo Chen 0007, Lingbing Guo, Huajun Chen |
ICLR | 2 |
| 2024 | Mol-Instructions: A Large-Scale Biomolecular Instruction Dataset for Large Language ModelsabstractLarge Language Models (LLMs), with their remarkable task-handling capabilities and innovative outputs, have catalyzed significant advancements across a spectrum of fields. However, their proficiency within specialized domains such as biomolecular studies remains limited. To address this challenge, we introduce Mol-Instructions, a comprehensive instruction dataset designed for the biomolecular domain. Mol-Instructions encompasses three key components: molecule-oriented instructions, protein-oriented instructions, and biomolecular text instructions. Each component aims to improve the understanding and prediction capabilities of LLMs concerning biomolecular features and behaviors. Through extensive instruction tuning experiments on LLMs, we demonstrate the effectiveness of Mol-Instructions in enhancing large models' performance in the intricate realm of biomolecular studies, thus fostering progress in the biomolecular research community. Mol-Instructions is publicly available for ongoing research and will undergo regular updates to enhance its applicability (https://github.com/zjunlp/Mol-Instructions). Yin Fang, Xiaozhuan Liang, Ningyu Zhang 0001, Kangwei Liu 0002, Zhuo Chen 0007, Huajun Chen |
ICLR | 3 |
| 2024 | Unveiling the Pitfalls of Knowledge Editing for Large Language ModelsabstractAs the cost associated with fine-tuning Large Language Models (LLMs) continues to rise, recent research efforts have pivoted towards developing methodologies to edit implicit knowledge embedded within LLMs. Yet, there's still a dark cloud lingering overhead -- will knowledge editing trigger butterfly effect? since it is still unclear whether knowledge editing might introduce side effects that pose potential risks or not. This paper pioneers the investigation into the potential pitfalls associated with knowledge editing for LLMs. To achieve this, we introduce new benchmark datasets and propose innovative evaluation metrics. Our results underline two pivotal concerns: (1) Knowledge Conflict: Editing groups of facts that logically clash can magnify the inherent inconsistencies in LLMs—a facet neglected by previous methods. (2) Knowledge Distortion: Altering parameters with the aim of editing factual knowledge can irrevocably warp the innate knowledge structure of LLMs. Experimental results vividly demonstrate that knowledge editing might inadvertently cast a shadow of unintended consequences on LLMs, which warrant attention and efforts for future works. Code and data are available at https://github.com/zjunlp/PitfallsKnowledgeEditing. Zhoubo Li, Ningyu Zhang 0001, Yunzhi Yao, Xi Chen 0003, Huajun Chen |
ICLR | 2 |
| 2024 | Neighboring Perturbations of Knowledge Editing on Large Language ModelsabstractDespite their exceptional capabilities, large language models (LLMs) are prone to generating unintended text due to false or outdated knowledge. Given the resource-intensive nature of retraining LLMs, there has been a notable increase in the development of knowledge editing. However, current approaches and evaluations rarely explore the perturbation of editing on neighboring knowledge. This paper studies whether updating new knowledge to LLMs perturbs the neighboring knowledge encapsulated within them. Specifically, we seek to figure out whether appending a new answer into an answer list to a factual question leads to catastrophic forgetting of original correct answers in this list, as well as unintentional inclusion of incorrect answers. A metric of additivity is introduced and a benchmark dubbed as Perturbation Evaluation of Appending Knowledge (PEAK) is constructed to evaluate the degree of perturbation to neighboring knowledge when appending new knowledge. Besides, a plug-and-play framework termed Appending via Preservation and Prevention (APP) is proposed to mitigate the neighboring perturbation by maintaining the integrity of the answer list. Experiments demonstrate the effectiveness of APP coupling with four editing methods on three LLMs. Jun-Yu Ma, Zhen-Hua Ling, Ningyu Zhang 0001, Jia-Chen Gu |
ICML | 3 |
| 2024 | InstructEdit: Instruction-Based Knowledge Editing for Large Language Models
Ningyu Zhang 0001, Bozhong Tian, Siyuan Cheng 0008, Xiaozhuan Liang, Kouying Xue, Yanjie Gou, Xi Chen 0003, Huajun Chen |
IJCAI | 1 |
| 2024 | FactCHD: Benchmarking Fact-Conflicting Hallucination Detection
Xiang Chen 0016, Duanzheng Song, Honghao Gui, Ningyu Zhang 0001, Yong Jiang 0005, Fei Huang 0002, Chengfei Lyu, Huajun Chen |
IJCAI | 5 |
| 2024 | Continual Multimodal Knowledge Graph Construction
Xiang Chen 0016, Jingtian Zhang, Ningyu Zhang 0001, Tongtong Wu, Yuxiang Wang 0001, Yongheng Wang, Huajun Chen |
IJCAI | 4 |
| 2024 | Knowledge-Informed Molecular Learning: A Survey on Paradigm Transfer
Yin Fang, Zhuo Chen 0007, Ningyu Zhang 0001, Huajun Chen |
KSEM (1) | 4 |
| 2024 | Making Language Models Better Tool Learners with Execution FeedbackabstractShuofei Qiao, Honghao Gui, Chengfei Lv, Qianghuai Jia, Huajun Chen, Ningyu Zhang. Proceedings of the 2024 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers). 2024. Shuofei Qiao, Honghao Gui, Chengfei Lv, Qianghuai Jia, Huajun Chen, Ningyu Zhang 0001 |
NAACL-HLT | 6 |
| 2024 | WISE: Rethinking the Knowledge Memory for Lifelong Model Editing of Large Language ModelsabstractLarge language models (LLMs) need knowledge updates to meet the ever-growing world facts and correct the hallucinated responses, facilitating the methods of lifelong model editing. Where the updated knowledge resides in memories is a fundamental question for model editing. In this paper, we find that editing either long-term memory (direct model parameters) or working memory (non-parametric knowledge of neural network activations/representations by retrieval) will result in an impossible triangle---reliability, generalization, and locality can not be realized together in the lifelong editing settings. For long-term memory, directly editing the parameters will cause conflicts with irrelevant pretrained knowledge or previous edits (poor reliability and locality). For working memory, retrieval-based activations can hardly make the model understand the edits and generalize (poor generalization). Therefore, we propose WISE to bridge the gap between memories. In WISE, we design a dual parametric memory scheme, which consists of the main memory for the pretrained knowledge and a side memory for the edited knowledge. We only edit the knowledge in the side memory and train a router to decide which memory to go through when given a query. For continual editing, we devise a knowledge-sharding mechanism where different sets of edits reside in distinct subspaces of parameters, and are subsequently merged into a shared memory without conflicts. Extensive experiments show that WISE can outperform previous model editing methods and overcome the impossible triangle under lifelong model editing of question answering, hallucination, and out-of-distribution settings across trending LLM architectures, e.g., GPT, LLaMA, and Mistral. Peng Wang 0104, Ningyu Zhang 0001, Ziwen Xu, Yunzhi Yao, Yong Jiang 0005, Pengjun Xie, Fei Huang 0002, Huajun Chen |
NeurIPS | 3 |
| 2024 | Agent Planning with World Knowledge ModelabstractRecent endeavors towards directly using large language models (LLMs) as agent models to execute interactive planning tasks have shown commendable results. Despite their achievements, however, they still struggle with brainless trial-and-error in global planning and generating hallucinatory actions in local planning due to their poor understanding of the "real" physical world. Imitating humans' mental world knowledge model which provides global prior knowledge before the task and maintains local dynamic knowledge during the task, in this paper, we introduce parametric World Knowledge Model (WKM) to facilitate agent planning. Concretely, we steer the agent model to self-synthesize knowledge from both expert and sampled trajectories. Then we develop WKM, providing prior task knowledge to guide the global planning and dynamic state knowledge to assist the local planning. Experimental results on three real-world simulated datasets with Mistral-7B, Gemma-7B, and Llama-3-8B demonstrate that our method can achieve superior performance compared to various strong baselines. Besides, we analyze to illustrate that our WKM can effectively alleviate the blind trial-and-error and hallucinatory action issues, providing strong support for the agent's understanding of the world. Other interesting findings include: 1) our instance-level task knowledge can generalize better to unseen tasks, 2) weak WKM can guide strong agent model planning, and 3) unified WKM training has promising potential for further development. Shuofei Qiao, Runnan Fang, Ningyu Zhang 0001, Xiang Chen 0016, Shumin Deng, Yong Jiang 0005, Pengjun Xie, Fei Huang 0002, Huajun Chen |
NeurIPS | 3 |
| 2024 | Knowledge Circuits in Pretrained TransformersabstractThe remarkable capabilities of modern large language models are rooted in their vast repositories of knowledge encoded within their parameters, enabling them to perceive the world and engage in reasoning. The inner workings of how these models store knowledge have long been a subject of intense interest and investigation among researchers. To date, most studies have concentrated on isolated components within these models, such as the Multilayer Perceptrons and attention head. In this paper, we delve into the computation graph of the language model to uncover the knowledge circuits that are instrumental in articulating specific knowledge. The experiments, conducted with GPT2 and TinyLLAMA, has allowed us to observe how certain information heads, relation heads, and Multilayer Perceptrons collaboratively encode knowledge within the model. Moreover, we evaluate the impact of current knowledge editing techniques on these knowledge circuits, providing deeper insights into the functioning and constraints of these editing methodologies. Finally, we utilize knowledge circuits to analyze and interpret language model behaviors such as hallucinations and in-context learning. We believe the knowledge circuit holds potential for advancing our understanding of Transformers and guiding the improved design of knowledge editing. Yunzhi Yao, Ningyu Zhang 0001, Zekun Xi, Ziwen Xu, Shumin Deng, Huajun Chen |
NeurIPS | 2 |
| 2024 | Editing Personality For Large Language Models
Shengyu Mao, Yong Jiang 0005, Pengjun Xie, Fei Huang 0002, Ningyu Zhang 0001 |
NLPCC (2) | 7 |
| 2024 | Overview of the NLPCC 2024 Shared Task 10: Regulating Large Language Models
Ziwen Xu, Xiang Chen 0016, Shumin Deng, Ningyu Zhang 0001 |
NLPCC (5) | 6 |
| 2024 | InstructIE: A Bilingual Instruction-based Information Extraction Dataset
Honghao Gui, Shuofei Qiao, Jintian Zhang, Hongbin Ye, Mengshu Sun, Lei Liang 0002, Jeff Z. Pan, Huajun Chen, Ningyu Zhang 0001 |
ISWC (3) | 9 |
| 2024 | A representation learning-based approach to enhancing manufacturing quality for low-voltage electrical products
Yuming Xu, Tao Peng 0012, Jiaqi Tao, Ao Bai, Ningyu Zhang 0001, Kendrik Lim |
Adv. Eng. Informatics | 5 |
| 2024 | Relphormer: Relational Graph Transformer for Knowledge Graph Representations
Zhen Bi, Siyuan Cheng 0008, Jing Chen 0060, Xiaozhuan Liang, Feiyu Xiong, Ningyu Zhang 0001 |
Neurocomputing | 6 |
| 2024 | CodeKGC: Code Language Model for Generative Knowledge Graph ConstructionabstractCurrent generative knowledge graph construction approaches usually fail to capture structural knowledge by simply flattening natural language into serialized texts or a specification language. However, large generative language model trained on structured data such as code has demonstrated impressive capability in understanding natural language for structural prediction and reasoning tasks. Intuitively, we address the task of generative knowledge graph construction with code language model: given a code-format natural language input, the target is to generate triples which can be represented as code completion tasks. Specifically, we develop schema-aware prompts that effectively utilize the semantic structure within the knowledge graph. As code inherently possesses structure, such as class and function definitions, it serves as a useful model for prior semantic structural knowledge. Furthermore, we employ a rationale-enhanced generation method to boost the performance. Rationales provide intermediate steps, thereby improving knowledge extraction abilities. Experimental results indicate that the proposed approach can obtain better performance on benchmark datasets compared with baselines. 1 Zhen Bi, Jing Chen 0060, Yinuo Jiang, Feiyu Xiong, Huajun Chen, Ningyu Zhang 0001 |
ACM Trans. Asian Low Resour. Lang. Inf. Process. | 7 |
| 2024 | Sequence Labeling as Non-Autoregressive Dual-Query Set GenerationabstractSequence labeling is a crucial task in the NLP community that aims at identifying and assigning spans within the input sentence. It has wide applications in various fields such as information extraction, dialogue system, and sentiment analysis. However, previously proposed span-based or sequence-to-sequence models conduct locating and assigning in order, resulting in problems of error propagation and unnecessary training loss, respectively. This paper addresses the problem by reformulating the sequence labeling as a non-autoregressive set generation to realize locating and assigning in parallel. Herein, we propose aDual-QuerySetGeneration (DQSetGen) model for unified sequence labeling tasks. Specifically, the dual-query set, including a prompted type query and a positional query with anchor span, is fed into the non-autoregressive decoder to probe the spans which correspond to the positional query and have similar patterns with the type query. By avoiding the autoregressive nature of previous approaches, our method significantly improves efficiency and reduces error propagation. Experimental results illustrate that our approach can obtain superior performance on 5 sub-tasks across 11 benchmark datasets. The non-autoregressive nature of our method allows for parallel computation, achieving faster inference speed than compared baselines. In conclusion, our proposed non-autoregressive dual-query set generation method offers a more efficient and accurate approach to sequence labeling tasks in NLP. Its advantages in terms of performance and efficiency make it a promising solution for various applications in data mining and other related fields. Xiang Chen 0016, Lei Li 0040, Shumin Deng, Chuanqi Tan, Fei Huang 0002, Luo Si, Ningyu Zhang 0001, Huajun Chen |
IEEE ACM Trans. Audio Speech Lang. Process. | 8 |
| 2024 | LLMs for knowledge graph construction and reasoning: recent capabilities and future opportunities
Jing Chen 0060, Shuofei Qiao, Yixin Ou, Yunzhi Yao, Shumin Deng, Huajun Chen, Ningyu Zhang 0001 |
World Wide Web (WWW) | 9 |
| 2023 | Multi-Modal Protein Knowledge Graph Construction and Applications (Student Abstract)abstractExisting data-centric methods for protein science generally cannot sufficiently capture and leverage biology knowledge, which may be crucial for many protein tasks. To facilitate research in this field, we create ProteinKG65, a knowledge graph for protein science. Using gene ontology and Uniprot knowledge base as a basis, we transform and integrate various kinds of knowledge with aligned descriptions and protein sequences, respectively, to GO terms and protein entities. ProteinKG65 is mainly dedicated to providing a specialized protein knowledge graph, bringing the knowledge of Gene Ontology to protein function and structure prediction. We also illustrate the potential applications of ProteinKG65 with a prototype. Our dataset can be downloaded at https://w3id.org/proteinkg65. Siyuan Cheng 0008, Xiaozhuan Liang, Zhen Bi, Huajun Chen, Ningyu Zhang 0001 |
AAAI | 5 |
| 2023 | On Analyzing the Role of Image for Visual-Enhanced Relation Extraction (Student Abstract)abstractMultimodal relation extraction is an essential task for knowledge graph construction. In this paper, we take an in-depth empirical analysis that indicates the inaccurate information in the visual scene graph leads to poor modal alignment weights, further degrading performance. Moreover, the visual shuffle experiments illustrate that the current approaches may not take full advantage of visual information. Based on the above observation, we further propose a strong baseline with an implicit fine-grained multimodal alignment based on Transformer for multimodal relation extraction. Experimental results demonstrate the better performance of our method. Codes are available at https://github.com/zjunlp/DeepKE/tree/main/example/re/multimodal. Lei Li 0040, Xiang Chen 0016, Shuofei Qiao, Feiyu Xiong, Huajun Chen, Ningyu Zhang 0001 |
AAAI | 6 |
| 2023 | SPEECH: Structured Prediction with Energy-Based Event-Centric HyperspheresabstractEvent-centric structured prediction involves predicting structured outputs of events.In most NLP cases, event structures are complex with manifold dependency, and it is challenging to effectively represent these complicated structured events.To address these issues, we propose Structured Prediction with Energybased Event-Centric Hyperspheres (SPEECH).SPEECH models complex dependency among event structured components with energybased modeling, and represents event classes with simple but effective hyperspheres.Experiments on two unified-annotated event datasets indicate that SPEECH is predominant in event detection and event-relation extraction tasks. Shumin Deng, Shengyu Mao, Ningyu Zhang 0001, Bryan Hooi |
ACL (1) | 3 |
| 2023 | Reasoning with Language Model Prompting: A SurveyabstractShuofei Qiao, Yixin Ou, Ningyu Zhang, Xiang Chen, Yunzhi Yao, Shumin Deng, Chuanqi Tan, Fei Huang, Huajun Chen. Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2023. Shuofei Qiao, Yixin Ou, Ningyu Zhang 0001, Xiang Chen 0016, Yunzhi Yao, Shumin Deng, Chuanqi Tan, Fei Huang 0002, Huajun Chen |
ACL (1) | 3 |
| 2023 | Can We Edit Multimodal Large Language Models?abstractIn this paper, we focus on editing Multimodal Large Language Models (MLLMs).Compared to editing single-modal LLMs, multimodal model editing is more challenging, which demands a higher level of scrutiny and careful consideration in the editing process.To facilitate research in this area, we construct a new benchmark, dubbed MMEdit, for editing multimodal LLMs and establishing a suite of innovative metrics for evaluation.We conduct comprehensive experiments involving various model editing baselines and analyze the impact of editing different components for multimodal LLMs.Empirically, we notice that previous baselines can implement editing multimodal LLMs to some extent, but the effect is still barely satisfactory, indicating the potential difficulty of this task.We hope that our work can provide the NLP community with insights1. Siyuan Cheng 0008, Bozhong Tian, Qingbin Liu, Xi Chen 0003, Yongheng Wang, Huajun Chen, Ningyu Zhang 0001 |
EMNLP | 7 |
| 2023 | Knowledge Rumination for Pre-trained Language ModelsabstractPrevious studies have revealed that vanilla pre-trained language models (PLMs) lack the capacity to handle knowledge-intensive NLP tasks alone; thus, several works have attempted to integrate external knowledge into PLMs.However, despite the promising outcome, we empirically observe that PLMs may have already encoded rich knowledge in their pre-trained parameters but fail to fully utilize them when applying them to knowledgeintensive tasks.In this paper, we propose a new paradigm dubbed Knowledge Rumination to help the pre-trained language model utilize that related latent knowledge without retrieving it from the external corpus.By simply adding a prompt like "As far as I know" to the PLMs, we try to review related latent knowledge and inject them back into the model for knowledge consolidation.We apply the proposed knowledge rumination to various language models, including RoBERTa, De-BERTa, and GPT-3.Experimental results on six commonsense reasoning tasks and GLUE benchmarks demonstrate the effectiveness of our proposed approach, which proves that the knowledge stored in PLMs can be better exploited to enhance performance 1 . Yunzhi Yao, Peng Wang 0104, Shengyu Mao, Chuanqi Tan, Fei Huang 0002, Huajun Chen, Ningyu Zhang 0001 |
EMNLP | 7 |
| 2023 | Editing Large Language Models: Problems, Methods, and OpportunitiesabstractDespite the ability to train capable LLMs, the methodology for maintaining their relevancy and rectifying errors remains elusive.To this end, the past few years have witnessed a surge in techniques for editing LLMs, the objective of which is to efficiently alter the behavior of LLMs within a specific domain without negatively impacting performance across other inputs.This paper embarks on a deep exploration of the problems, methods, and opportunities related to model editing for LLMs.In particular, we provide an exhaustive overview of the task definition and challenges associated with model editing, along with an in-depth empirical analysis of the most progressive methods currently at our disposal.We also build a new benchmark dataset to facilitate a more robust evaluation and pinpoint enduring issues intrinsic to existing techniques.Our objective is to provide valuable insights into the effectiveness and feasibility of each editing technique, thereby assisting the community in making informed decisions on the selection of the most appropriate method for a specific task or context 1 . Yunzhi Yao, Peng Wang 0104, Bozhong Tian, Siyuan Cheng 0008, Zhoubo Li, Shumin Deng, Huajun Chen, Ningyu Zhang 0001 |
EMNLP | 8 |
| 2023 | Construction and Applications of Billion-Scale Pre-Trained Multimodal Business Knowledge GraphabstractBusiness Knowledge Graphs (KGs) are important to many enterprises today, providing factual knowledge and structured data that steer many products and make them more intelligent. Despite their promising benefits, building business KG necessitates solving prohibitive issues of deficient structure and multiple modalities. In this paper, we advance the understanding of the practical challenges related to building KG in non-trivial real-world systems. We introduce the process of building an open business knowledge graph (OpenBG) derived from a well-known enterprise, Alibaba Group. Specifically, we define a core ontology to cover various abstract products and consumption demands, with fine-grained taxonomy and multimodal facts in deployed applications. OpenBG is an open business KG of unprecedented scale: 2.6 billion triples with more than 88 million entities covering over 1 million core classes/concepts and 2,681 types of relations. We release all the open resources (OpenBG benchmarks) derived from it for the community and report experimental results of KG-centric tasks. We also run up an online competition based on OpenBG benchmarks, and has attracted thousands of teams. We further pre-train OpenBG and apply it to many KG-enhanced downstream tasks in business scenarios, demonstrating the effectiveness of billion-scale multimodal knowledge for e-commerce. All the resources with codes have been released at https://github.com/OpenBGBenchmark/OpenBG. Shumin Deng, Zhoubo Li, Ningyu Zhang 0001, Zelin Dai, Hehong Chen, Feiyu Xiong, Ming Yan 0008, Mosha Chen, Jiaoyan Chen 0001, Jeff Z. Pan, Bryan Hooi, Huajun Chen |
ICDE | 4 |
| 2023 | Multimodal Analogical Reasoning over Knowledge Graphs
Ningyu Zhang 0001, Lei Li 0040, Xiang Chen 0016, Xiaozhuan Liang, Shumin Deng, Huajun Chen |
ICLR | 1 |
| 2023 | One Model for All Domains: Collaborative Domain-Prefix Tuning for Cross-Domain NERabstractCross-domain NER is a challenging task to address the low-resource problem in practical scenarios. Previous typical solutions mainly obtain a NER model by pre-trained language models (PLMs) with data from a rich-resource domain and adapt it to the target domain. Owing to the mismatch issue among entity types in different domains, previous approaches normally tune all parameters of PLMs, ending up with an entirely new NER model for each domain. Moreover, current models only focus on leveraging knowledge in one general source domain while failing to successfully transfer knowledge from multiple sources to the target. To address these issues, we introduce Collaborative Domain-Prefix Tuning for cross-domain NER (CP-NER) based on text-to-text generative PLMs. Specifically, we present text-to-text generation grounding domain-related instructors to transfer knowledge to new domain NER tasks without structural modifications. We utilize frozen PLMs and conduct collaborative domain-prefix tuning to stimulate the potential of PLMs to handle NER tasks across various domains. Experimental results on the Cross-NER benchmark show that the proposed approach has flexible transfer ability and performs better on both one-source and multiple-source cross-domain NER tasks. Xiang Chen 0016, Lei Li 0040, Shuofei Qiao, Ningyu Zhang 0001, Chuanqi Tan, Yong Jiang 0005, Fei Huang 0002, Huajun Chen |
IJCAI | 4 |
| 2023 | CAMVR: Context-Adaptive Multi-View Representation Learning for Dense RetrievalabstractThe recently proposed MVR (Multi-View Repre-sentation) model achieves remarkable performance for open-domain dense retrieval. In MVR, matching multi-view queries is enabled by encoding the document into multiple representations. However, these representations tend to collapse into the same one when the percentage of documents answering multiple queries in the training data is low. Here, we propose a Context-Adaptive Multi-View Representation (CAMVR) learning framework that explicitly avoids this collapse problem by aligning each viewer token with different document snippets. In CAMVR, each viewer token is placed before a snippet to capture the local and global information with consideration that answers of different view queries may scatter in one document. The view of the snippet containing the answer is also used to explicitly supervise the learning process, from which the interpretability of the view representation is provided. Lastly, this work presents extensive experiments demonstrating that CAMVR outperforms the existing models to achieve state-of-the-art results. Zhilin Liao, Xiangting Hou, Dongfang Lou, Ningyu Zhang 0001 |
IJCNN | 5 |
| 2023 | Newton-Cotes Graph Neural Networks: On the Time Evolution of Dynamic SystemsabstractReasoning system dynamics is one of the most important analytical approaches for many scientific studies. With the initial state of a system as input, the recent graph neural networks (GNNs)-based methods are capable of predicting the future state distant in time with high accuracy. Although these methods have diverse designs in modeling the coordinates and interacting forces of the system, we show that they actually share a common paradigm that learns the integration of the velocity over the interval between the initial and terminal coordinates. However, their integrand is constant w.r.t. time. Inspired by this observation, we propose a new approach to predict the integration based on several velocity estimations with Newton–Cotes formulas and prove its effectiveness theoretically. Extensive experiments on several benchmarks empirically demonstrate consistent and significant improvement compared with the state-of-the-art methods. Lingbing Guo, Weiqing Wang 0001, Zhuo Chen 0007, Ningyu Zhang 0001, Zequn Sun 0001, Yixuan Lai, Qiang Zhang 0026, Huajun Chen |
NeurIPS | 4 |
| 2023 | Reasoning Through Memorization: Nearest Neighbor Knowledge Graph Embeddings
Peng Wang 0104, Xin Xie 0006, Ningyu Zhang 0001 |
NLPCC (1) | 4 |
| 2023 | Schema-aware Reference as Prompt Improves Data-Efficient Knowledge Graph ConstructionabstractWith the development of pre-trained language models, many prompt-based approaches to data-efficient knowledge graph construction have been proposed and achieved impressive performance. However, existing prompt-based learning methods for knowledge graph construction are still susceptible to several potential limitations: (i) semantic gap between natural language and output structured knowledge with pre-defined schema, which means model cannot fully exploit semantic knowledge with the constrained templates; (ii) representation learning with locally individual instances limits the performance given the insufficient features, which are unable to unleash the potential analogical capability of pre-trained language models. Motivated by these observations, we propose a retrieval-augmented approach, which retrieves schema-aware Reference As Prompt (RAP), for data-efficient knowledge graph construction. It can dynamically leverage schema and knowledge inherited from human-annotated and weak-supervised data as a prompt for each sample, which is model-agnostic and can be plugged into widespread existing approaches. Experimental results demonstrate that previous methods integrated with RAP can achieve impressive performance gains in low-resource settings on five datasets of relational triple extraction and event extraction for knowledge graph construction Code is available in https://github.com/zjunlp/RAP. Yunzhi Yao, Shengyu Mao, Ningyu Zhang 0001, Xiang Chen 0016, Shumin Deng, Xi Chen 0003, Huajun Chen |
SIGIR | 3 |
| 2023 | LOGEN: Few-Shot Logical Knowledge-Conditioned Text Generation With Self-TrainingabstractNatural language generation from structured data mainly focuses on surface-level descriptions, suffering from uncontrollable content selection and low fidelity. Previous works leverage logical forms to facilitate logical knowledge-conditioned text generation. Though achieving remarkable progress, they are data-hungry, which makes the adoption for real-world applications challenging with limited data. To this end, this paper proposes a unified framework for logical knowledge-conditioned text generation in the few-shot setting. With only a few seeds logical forms (e.g., 20/100 shot), our approach leverages self-training and samples pseudo logical forms based on content and structure consistency. Experimental results demonstrate that our approach can obtain better few-shot performance than baselines. Shumin Deng, Hongbin Ye, Chuanqi Tan, Mosha Chen, Songfang Huang, Fei Huang 0002, Huajun Chen, Ningyu Zhang 0001 |
IEEE ACM Trans. Audio Speech Lang. Process. | 9 |
| 2022 | Learning to Ask for Data-Efficient Event Argument Extraction (Student Abstract)abstractEvent argument extraction (EAE) is an important task for information extraction to discover specific argument roles. In this study, we cast EAE as a question-based cloze task and empirically analyze fixed discrete token template performance. As generating human-annotated question templates is often time-consuming and labor-intensive, we further propose a novel approach called “Learning to Ask,” which can learn optimized question templates for EAE without human annotations. Experiments using the ACE-2005 dataset demonstrate that our method based on optimized questions achieves state-of-the-art performance in both the few-shot and supervised settings. Hongbin Ye, Ningyu Zhang 0001, Zhen Bi, Shumin Deng, Chuanqi Tan, Hui Chen 0018, Fei Huang 0002, Huajun Chen |
AAAI | 2 |
| 2022 | CBLUE: A Chinese Biomedical Language Understanding Evaluation BenchmarkabstractNingyu Zhang, Mosha Chen, Zhen Bi, Xiaozhuan Liang, Lei Li, Xin Shang, Kangping Yin, Chuanqi Tan, Jian Xu, Fei Huang, Luo Si, Yuan Ni, Guotong Xie, Zhifang Sui, Baobao Chang, Hui Zong, Zheng Yuan, Linfeng Li, Jun Yan, Hongying Zan, Kunli Zhang, Buzhou Tang, Qingcai Chen. Proceedings of the 60th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2022. Ningyu Zhang 0001, Mosha Chen, Zhen Bi, Xiaozhuan Liang, Lei Li 0040, Xin Shang, Kangping Yin, Chuanqi Tan, Fei Huang 0002, Luo Si, Yuan Ni, Guo Tong Xie, Zhifang Sui, Baobao Chang, Hui Zong, Zheng Yuan 0002, Jun Yan 0010, Hongying Zan, Kunli Zhang, Buzhou Tang, Qingcai Chen |
ACL (1) | 1 |
| 2022 | Finding Influential Instances for Distantly Supervised Relation ExtractionabstractDistant supervision (DS) is a strong way to expand the datasets for enhancing relation extraction (RE) models but often suffers from high label noise. Current works based on attention, reinforcement learning, or GAN are black-box models so they neither provide meaningful interpretation of sample selection in DS nor stability on different domains. On the contrary, this work proposes a novel model-agnostic instance sampling method for DS by influence function (IF), namely REIF. Our method identifies favorable/unfavorable instances in the bag based on IF, then does dynamic instance sampling. We design a fast influence sampling algorithm that reduces the computational complexity from \mathcal{O}(mn) to \mathcal{O}(1), with analyzing its robustness on the selected sampling function. Experiments show that by simply sampling the favorable instances during training, REIF is able to win over a series of baselines which have complicated architectures. We also demonstrate that REIF can support interpretable instance selection. Zifeng Wang 0008, Rui Wen 0001, Xi Chen 0003, Shao-Lun Huang, Ningyu Zhang 0001, Yefeng Zheng 0001 |
COLING | 5 |
| 2022 | LightNER: A Lightweight Tuning Paradigm for Low-resource NER via Pluggable PromptingabstractMost NER methods rely on extensive labeled data for model training, which struggles in the low-resource scenarios with limited training data. Existing dominant approaches usually suffer from the challenge that the target domain has different label sets compared with a resource-rich source domain, which can be concluded as class transfer and domain transfer. In this paper, we propose a lightweight tuning paradigm for low-resource NER via pluggable prompting (LightNER). Specifically, we construct the unified learnable verbalizer of entity categories to generate the entity span sequence and entity categories without any label-specific classifiers, thus addressing the class transfer issue. We further propose a pluggable guidance module by incorporating learnable parameters into the self-attention layer as guidance, which can re-modulate the attention and adapt pre-trained weights. Note that we only tune those inserted module with the whole parameter of the pre-trained language model fixed, thus, making our approach lightweight and flexible for low-resource scenarios and can better transfer knowledge across domains. Experimental results show that LightNER can obtain comparable performance in the standard supervised setting and outperform strong baselines in low-resource settings. Xiang Chen 0016, Lei Li 0040, Shumin Deng, Chuanqi Tan, Changliang Xu, Fei Huang 0002, Luo Si, Huajun Chen, Ningyu Zhang 0001 |
COLING | 9 |
| 2022 | Generative Knowledge Graph Construction: A ReviewabstractGenerative Knowledge Graph Construction (KGC) refers to those methods that leverage the sequence-to-sequence framework for building knowledge graphs, which is flexible and can be adapted to widespread tasks.In this study, we summarize the recent compelling progress in generative knowledge graph construction.We present the advantages and weaknesses of each paradigm in terms of different generation targets and provide theoretical insight and empirical analysis.Based on the review, we suggest promising research directions for the future.Our contributions are threefold: (1) We present a detailed, complete taxonomy for the generative KGC methods; (2) We provide a theoretical and empirical analysis of the generative KGC methods; (3) We propose several research directions that can be developed in the future. Hongbin Ye, Ningyu Zhang 0001, Hui Chen 0018, Huajun Chen |
EMNLP | 2 |
| 2022 | OntoProtein: Protein Pretraining With Gene Ontology Embedding
Ningyu Zhang 0001, Zhen Bi, Xiaozhuan Liang, Siyuan Cheng 0008, Haosen Hong, Shumin Deng, Qiang Zhang 0026, Jiazhang Lian, Huajun Chen |
ICLR | 1 |
| 2022 | Differentiable Prompt Makes Pre-trained Language Models Better Few-shot Learners
Ningyu Zhang 0001, Luoqiu Li, Xiang Chen 0016, Shumin Deng, Zhen Bi, Chuanqi Tan, Fei Huang 0002, Huajun Chen |
ICLR | 1 |
| 2022 | Shifting Perspective to See Difference: A Novel Multi-view Method for Skeleton based Action RecognitionabstractSkeleton-based human action recognition is a longstanding challenge due to its complex dynamics. Some fine-grain details of the dynamics play a vital role in classification. The existing work largely focuses on designing incremental neural networks with more complicated adjacent matrices to capture the details of joints relationships. However, they still have difficulties distinguishing actions that have broadly similar motion patterns but belong to different categories. Interestingly, we found that the subtle differences in motion patterns can be significantly amplified and become easy for audience to distinct through specified view directions, where this property haven't been fully explored before. Drastically different from previous work, we boost the performance by proposing a conceptually simple yet effective Multi-view strategy that recognizes actions from a collection of dynamic view features. Specifically, we design a novel Skeleton-Anchor Proposal (SAP) module which contains a Multi-head structure to learn a set of views. For feature learning of different views, we introduce a novel Angle Representation to transform the actions under different views and feed the transformations into the baseline model. Our module can work seamlessly with the existing action classification model. Incorporated with baseline models, our SAP module exhibits clear performance gains on many challenging benchmarks. Moreover, comprehensive experiments show that our model consistently beats down the state-of-the-art and remains effective and robust especially when dealing with corrupted data. Related code will be available on https://github.com/ideal-idea/SAP Ruijie Hou, Yanran Li, Ningyu Zhang 0001, Xiaosong Yang |
ACM Multimedia | 3 |
| 2022 | Decoupling Knowledge from Memorization: Retrieval-augmented Prompt LearningabstractPrompt learning approaches have made waves in natural language processing by inducing better few-shot performance while they still follow a parametric-based learning paradigm; the oblivion and rote memorization problems in learning may encounter unstable generalization issues. Specifically, vanilla prompt learning may struggle to utilize atypical instances by rote during fully-supervised training or overfit shallow patterns with low-shot data. To alleviate such limitations, we develop RetroPrompt with the motivation of decoupling knowledge from memorization to help the model strike a balance between generalization and memorization. In contrast with vanilla prompt learning, RetroPrompt constructs an open-book knowledge-store from training instances and implements a retrieval mechanism during the process of input, training and inference, thus equipping the model with the ability to retrieve related contexts from the training corpus as cues for enhancement. Extensive experiments demonstrate that RetroPrompt can obtain better performance in both few-shot and zero-shot settings. Besides, we further illustrate that our proposed RetroPrompt can yield better generalization abilities with new datasets. Detailed analysis of memorization indeed reveals RetroPrompt can reduce the reliance of language models on memorization; thus, improving generalization for downstream tasks. Code is available in https://github.com/zjunlp/PromptKG/tree/main/research/RetroPrompt. Xiang Chen 0016, Lei Li 0040, Ningyu Zhang 0001, Xiaozhuan Liang, Shumin Deng, Chuanqi Tan, Fei Huang 0002, Luo Si, Huajun Chen |
NeurIPS | 3 |
| 2022 | Kformer: Knowledge Injection in Transformer Feed-Forward Layers
Yunzhi Yao, Shaohan Huang, Li Dong 0004, Furu Wei, Huajun Chen, Ningyu Zhang 0001 |
NLPCC (1) | 6 |
| 2022 | Relation Extraction as Open-book Examination: Retrieval-enhanced Prompt TuningabstractPre-trained language models have contributed significantly to relation extraction by demonstrating remarkable few-shot learning abilities. However, prompt tuning methods for relation extraction may still fail to generalize to those rare or hard patterns. Note that the previous parametric learning paradigm can be viewed as memorization regarding training data as a book and inference as the close-book test. Those long-tailed or hard patterns can hardly be memorized in parameters given few-shot instances. To this end, we regard RE as an open-book examination and propose a new semiparametric paradigm of retrieval-enhanced prompt tuning for relation extraction. We construct an open-book datastore for retrieval regarding prompt-based instance representations and corresponding relation labels as memorized key-value pairs. During inference, the model can infer relations by linearly interpolating the base output of PLM with the non-parametric nearest neighbor distribution over the datastore. In this way, our model not only infers relation through knowledge stored in the weights during training but also assists decision-making by unwinding and querying examples in the open-book datastore. Extensive experiments on benchmark datasets show that our method can achieve state-of-the-art in both standard supervised and few-shot settings Xiang Chen 0016, Lei Li 0040, Ningyu Zhang 0001, Chuanqi Tan, Fei Huang 0002, Luo Si, Huajun Chen |
SIGIR | 3 |
| 2022 | Hybrid Transformer with Multi-level Fusion for Multimodal Knowledge Graph CompletionabstractMultimodal Knowledge Graphs (MKGs), which organize visual-text factual knowledge, have recently been successfully applied to tasks such as information retrieval, question answering, and recommendation system. Since most MKGs are far from complete, extensive knowledge graph completion studies have been proposed focusing on the multimodal entity, relation extraction and link prediction. However, different tasks and modalities require changes to the model architecture, and not all images/objects are relevant to text input, which hinders the applicability to diverse real-world scenarios. In this paper, we propose a hybrid transformer with multi-level fusion to address those issues. Specifically, we leverage a hybrid transformer architecture with unified input-output for diverse multimodal knowledge graph completion tasks. Moreover, we propose multi-level fusion, which integrates visual and text representation via coarse-grained prefix-guided interaction and fine-grained correlation-aware fusion modules. We conduct extensive experiments to validate that our MKGformer can obtain SOTA performance on four datasets of multimodal link prediction, multimodal RE, and multimodal NER1. https://github.com/zjunlp/MKGformer. Xiang Chen 0016, Ningyu Zhang 0001, Lei Li 0040, Shumin Deng, Chuanqi Tan, Changliang Xu, Fei Huang 0002, Luo Si, Huajun Chen |
SIGIR | 2 |
| 2022 | NeuralKG: An Open Source Library for Diverse Representation Learning of Knowledge GraphsabstractNeuralKG is an open-source Python-based library for diverse representation learning of knowledge graphs. It implements three kinds of Knowledge Graph Embedding (KGE) methods, including conventional KGEs, GNN-based KGEs, and Rule-based KGEs. With a unified framework, NeuralKG successfully reproduces link prediction results of these methods on benchmarks, freeing users from the laborious task of reimplementing them, especially for some methods originally written in non-python programming languages. Besides, NeuralKG is highly configurable and extensible. It provides various decoupled modules that can be mixed and adapted to each other. Thus with NeuralKG, developers and researchers can quickly implement their own designed models and obtain the optimal training methods to achieve the best performance efficiently. We built a website http://neuralkg.zjukg.org to organize an open and shared KG representation learning community. The library, experimental methodologies, and model reimplement results of NeuralKG are all publicly released at https://github.com/zjukg/NeuralKG. Wen Zhang 0015, Xiangnan Chen, Zhen Yao 0001, Mingyang Chen 0002, Yushan Zhu, Ningyu Zhang 0001, Zezhong Xu, Zonggang Yuan, Feiyu Xiong, Huajun Chen |
SIGIR | 9 |
| 2022 | KnowPrompt: Knowledge-aware Prompt-tuning with Synergistic Optimization for Relation ExtractionabstractRecently, prompt-tuning has achieved promising results for specific few-shot classification tasks. The core idea of prompt-tuning is to insert text pieces (i.e., templates) into the input and transform a classification task into a masked language modeling problem. However, for relation extraction, determining an appropriate prompt template requires domain expertise, and it is cumbersome and time-consuming to obtain a suitable label word. Furthermore, there exists abundant semantic and prior knowledge among the relation labels that cannot be ignored. To this end, we focus on incorporating knowledge among relation labels into prompt-tuning for relation extraction and propose a Knowledge-aware Prompt-tuning approach with synergistic optimization (KnowPrompt). Specifically, we inject latent knowledge contained in relation labels into prompt construction with learnable virtual type words and answer words. Then, we synergistically optimize their representation with structured constraints. Extensive experimental results on five datasets with standard and low-resource settings demonstrate the effectiveness of our approach. Our code and datasets are available in GitHub1 for reproducibility. Xiang Chen 0016, Ningyu Zhang 0001, Xin Xie 0006, Shumin Deng, Yunzhi Yao, Chuanqi Tan, Fei Huang 0002, Luo Si, Huajun Chen |
WWW | 2 |
| 2022 | Ontology-enhanced Prompt-tuning for Few-shot LearningabstractFew-shot Learning (FSL) is aimed to make predictions based on a limited number of samples. Structured data such as knowledge graphs and ontology libraries has been leveraged to benefit the few-shot setting in various tasks. However, the priors adopted by the existing methods suffer from challenging knowledge missing, knowledge noise, and knowledge heterogeneity, which hinder the performance for few-shot learning. In this study, we explore knowledge injection for FSL with pre-trained language models and propose ontology-enhanced prompt-tuning (OntoPrompt). Specifically, we develop the ontology transformation based on the external knowledge graph to address the knowledge missing issue, which fulfills and converts structure knowledge to text. We further introduce span-sensitive knowledge injection via a visible matrix to select informative knowledge to handle the knowledge noise issue. To bridge the gap between knowledge and text, we propose a collective training algorithm to optimize representations jointly. We evaluate our proposed OntoPrompt in three tasks, including relation extraction, event extraction, and knowledge graph completion, with eight datasets. Experimental results demonstrate that our approach can obtain better few-shot performance than baselines. Hongbin Ye, Ningyu Zhang 0001, Shumin Deng, Xiang Chen 0016, Hui Chen 0018, Feiyu Xiong, Xi Chen 0003, Huajun Chen |
WWW | 2 |
| 2022 | Robust triple extraction with cascade bidirectional capsule network
Ningyu Zhang 0001, Shumin Deng, Hongbin Ye, Wei Zhang 0127, Huajun Chen |
Expert Syst. Appl. | 1 |
| 2022 | Low-resource extraction with knowledge-aware pairwise prototype learning
Shumin Deng, Ningyu Zhang 0001, Hui Chen 0018, Chuanqi Tan, Fei Huang 0002, Changliang Xu, Huajun Chen |
Knowl. Based Syst. | 2 |
| 2021 | Contrastive Triple Extraction with Generative TransformerabstractTriple extraction is an essential task in information extraction for natural language processing and knowledge graph construction. In this paper, we revisit the end-to-end triple extraction task for sequence generation. Since generative triple extraction may struggle to capture long-term dependencies and generate unfaithful triples, we introduce a novel model, contrastive triple extraction with a generative transformer. Specifically, we introduce a single shared transformer module for encoder-decoder-based generation. To generate faithful results, we propose a novel triplet contrastive training object. Moreover, we introduce two mechanisms to further improve model performance (i.e., batch-wise dynamic attention-masking and triple-wise calibration). Experimental results on three datasets (i.e., NYT, WebNLG, and MIE) show that our approach achieves better performance than that of baselines. Hongbin Ye, Ningyu Zhang 0001, Shumin Deng, Mosha Chen, Chuanqi Tan, Fei Huang 0002, Huajun Chen |
AAAI | 2 |
| 2021 | OntoED: Low-resource Event Detection with Ontology EmbeddingabstractShumin Deng, Ningyu Zhang, Luoqiu Li, Chen Hui, Tou Huaixiao, Mosha Chen, Fei Huang, Huajun Chen. Proceedings of the 59th Annual Meeting of the Association for Computational Linguistics and the 11th International Joint Conference on Natural Language Processing (Volume 1: Long Papers). 2021. Shumin Deng, Ningyu Zhang 0001, Luoqiu Li, Hui Chen 0018, Huaixiao Tou, Mosha Chen, Fei Huang 0002, Huajun Chen |
ACL/IJCNLP (1) | 2 |
| 2021 | MLBiNet: A Cross-Sentence Collective Event Detection NetworkabstractDongfang Lou, Zhilin Liao, Shumin Deng, Ningyu Zhang, Huajun Chen. Proceedings of the 59th Annual Meeting of the Association for Computational Linguistics and the 11th International Joint Conference on Natural Language Processing (Volume 1: Long Papers). 2021. Dongfang Lou, Zhilin Liao, Shumin Deng, Ningyu Zhang 0001, Huajun Chen |
ACL/IJCNLP (1) | 4 |
| 2021 | PRGC: Potential Relation and Global Correspondence Based Joint Relational Triple ExtractionabstractHengyi Zheng, Rui Wen, Xi Chen, Yifan Yang, Yunyan Zhang, Ziheng Zhang, Ningyu Zhang, Bin Qin, Xu Ming, Yefeng Zheng. Proceedings of the 59th Annual Meeting of the Association for Computational Linguistics and the 11th International Joint Conference on Natural Language Processing (Volume 1: Long Papers). 2021. Hengyi Zheng, Rui Wen 0001, Xi Chen 0003, Yifan Yang 0008, Yunyan Zhang, Ningyu Zhang 0001, Xu Ming, Yefeng Zheng 0001 |
ACL/IJCNLP (1) | 7 |
| 2021 | Unsupervised Knowledge Graph Alignment by Probabilistic Reasoning and Semantic EmbeddingabstractKnowledge Graph (KG) alignment is to discover the mappings (i.e., equivalent entities, relations, and others) between two KGs. The existing methods can be divided into the embedding-based models, and the conventional reasoning and lexical matching based systems. The former compute the similarity of entities via their cross-KG embeddings, but they usually rely on an ideal supervised learning setting for good performance and lack appropriate reasoning to avoid logically wrong mappings; while the latter address the reasoning issue but are poor at utilizing the KG graph structures and the entity contexts. In this study, we aim at combining the above two solutions and thus propose an iterative framework named PRASE which is based on probabilistic reasoning and semantic embedding. It learns the KG embeddings via entity mappings from a probabilistic reasoning system named PARIS, and feeds the resultant entity mappings and embeddings back into PARIS for augmentation. The PRASE framework is compatible with different embedding-based models, and our experiments on multiple datasets have demonstrated its state-of-the-art performance. Jiaoyan Chen 0001, Xi Chen 0003, Yuejia Xiang, Ningyu Zhang 0001, Yefeng Zheng 0001 |
IJCAI | 6 |
| 2021 | Document-level Relation Extraction as Semantic SegmentationabstractDocument-level relation extraction aims to extract relations among multiple entity pairs from a document. Previously proposed graph-based or transformer-based models utilize the entities independently, regardless of global information among relational triples. This paper approaches the problem by predicting an entity-level relation matrix to capture local and global information, parallel to the semantic segmentation task in computer vision. Herein, we propose a Document U-shaped Network for document-level relation extraction. Specifically, we leverage an encoder module to capture the context information of entities and a U-shaped segmentation module over the image-style feature map to capture global interdependency among triples. Experimental results show that our approach can obtain state-of-the-art performance on three benchmark datasets DocRED, CDR, and GDA. Ningyu Zhang 0001, Xiang Chen 0016, Xin Xie 0006, Shumin Deng, Chuanqi Tan, Mosha Chen, Fei Huang 0002, Luo Si, Huajun Chen |
IJCAI | 1 |
| 2021 | Drop Redundant, Shrink Irrelevant: Selective Knowledge Injection for Language PretrainingabstractPrevious research has demonstrated the power of leveraging prior knowledge to improve the performance of deep models in natural language processing. However, traditional methods neglect the fact that redundant and irrelevant knowledge exists in external knowledge bases. In this study, we launched an in-depth empirical investigation into downstream tasks and found that knowledge-enhanced approaches do not always exhibit satisfactory improvements. To this end, we investigate the fundamental reasons for ineffective knowledge infusion and present selective injection for language pretraining, which constitutes a model-agnostic method and is readily pluggable into previous approaches. Experimental results on benchmark datasets demonstrate that our approach can enhance state-of-the-art knowledge injection methods. Ningyu Zhang 0001, Shumin Deng, Xi Chen 0003, Yichi Zhang 0010, Wei Zhang 0127, Huajun Chen |
IJCAI | 1 |
| 2021 | AliCG: Fine-grained and Evolvable Conceptual Graph Construction for Semantic Search at AlibabaabstractConceptual graphs, which is a particular type of Knowledge Graphs, play an essential role in semantic search. Prior conceptual graph construction approaches typically extract high-frequent, coarse-grained, and time-invariant concepts from formal texts such as Wikipedia. In real applications, however, it is necessary to extract less-frequent, fine-grained, and time-varying conceptual knowledge and build taxonomy in an evolving manner. In this paper, we introduce an approach to implementing and deploying the conceptual graph at Alibaba. Specifically, We propose a framework called AliCG which is capable of a) extracting fine-grained concepts by a novel bootstrapping with alignment consensus approach, b) mining long-tail concepts with a novel low-resource phrase mining approach, c) updating the graph dynamically via a concept distribution estimation method based on implicit and explicit user behaviors. We have deployed the conceptual graph at Alibaba UC Browser. Extensive offline evaluation as well as online A/B testing demonstrate the efficacy of our approach. Ningyu Zhang 0001, Qianghuai Jia, Shumin Deng, Xiang Chen 0016, Hongbin Ye, Hui Chen 0018, Huaixiao Tou, Gang Huang 0004, Nengwei Hua, Huajun Chen |
KDD | 1 |
| 2021 | Knowledge Perceived Multi-modal Pretraining in E-commerceabstractIn this paper, we address multi-modal pretraining of product data in the field of E-commerce. Current multi-modal pretraining methods proposed for image and text modalities lack robustness in the face of modality-missing and modality-noise, which are two pervasive problems of multi-modal product data in real E-commerce scenarios. To this end, we propose a novel method, K3M, which introduces knowledge modality in multi-modal pretraining to correct the noise and supplement the missing of image and text modalities. The modal-encoding layer extracts the features of each modality. The modal-interaction layer is capable of effectively modeling the interaction of multiple modalities, where an initial-interactive feature fusion model is designed to maintain the independence of image modality and text modality, and a structure aggregation module is designed to fuse the information of image, text, and knowledge modalities. We pretrain K3M with three pretraining tasks, including masked object modeling (MOM), masked language modeling (MLM), and link prediction modeling (LPM). Experimental results on a real-world E-commerce dataset and a series of product-based downstream tasks demonstrate that K3M achieves significant improvements in performances than the baseline and state-of-the-art methods when modality-noise or modality-missing exists. Yushan Zhu, Huaixiao Zhao, Wen Zhang 0015, Ganqiang Ye, Hui Chen 0018, Ningyu Zhang 0001, Huajun Chen |
ACM Multimedia | 6 |
| 2021 | Noisy-Labeled NER with Confidence EstimationabstractKun Liu, Yao Fu, Chuanqi Tan, Mosha Chen, Ningyu Zhang, Songfang Huang, Sheng Gao. Proceedings of the 2021 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies. 2021. Chuanqi Tan, Mosha Chen, Ningyu Zhang 0001, Songfang Huang |
NAACL-HLT | 5 |
| 2021 | Field Embedding: A Unified Grain-Based Framework for Word RepresentationabstractJunjie Luo, Xi Chen, Jichao Sun, Yuejia Xiang, Ningyu Zhang, Xiang Wan. Proceedings of the 2021 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies. 2021. Junjie Luo 0003, Xi Chen 0003, Jichao Sun, Yuejia Xiang, Ningyu Zhang 0001 |
NAACL-HLT | 5 |
| 2021 | Contrastive Information Extraction With Generative TransformerabstractInformation extraction tasks such as triple extraction and event extraction are of great importance for natural language processing and knowledge graph construction. In this paper, we revisit the end-to-end information extraction task for sequence generation. Since generative information extraction may struggle to capture long-term dependencies and generate unfaithful triples, we introduce a novel model, contrastive information extraction with a generative transformer. Specifically, we introduce a single shared transformer module for an encoder-decoder-based generation. To generate faithful results, we propose a novel triplet contrastive training object. Moreover, we introduce two mechanisms to further improve model performance (i.e., batch-wise dynamic attention-masking and triple-wise calibration). Experimental results on five datasets (i.e., NYT, WebNLG, MIE, ACE-2005, and MUC-4) show that our approach achieves better performance than baselines. Ningyu Zhang 0001, Hongbin Ye, Shumin Deng, Chuanqi Tan, Mosha Chen, Songfang Huang, Fei Huang 0002, Huajun Chen |
IEEE ACM Trans. Audio Speech Lang. Process. | 1 |
| 2020 | When Low Resource NLP Meets Unsupervised Language Model: Meta-Pretraining then Meta-Learning for Few-Shot Text Classification (Student Abstract)abstractText classification tends to be difficult when data are deficient or when it is required to adapt to unseen classes. In such challenging scenarios, recent studies have often used meta-learning to simulate the few-shot task, thus negating implicit common linguistic features across tasks. This paper addresses such problems using meta-learning and unsupervised language models. Our approach is based on the insight that having a good generalization from a few examples relies on both a generic model initialization and an effective strategy for adapting this model to newly arising tasks. We show that our approach is not only simple but also produces a state-of-the-art performance on a well-studied sentiment classification dataset. It can thus be further suggested that pretraining could be a promising solution for few-shot learning of many other NLP tasks. The code and the dataset to replicate the experiments are made available at https://github.com/zxlzr/FewShotNLP. Shumin Deng, Ningyu Zhang 0001, Zhanlin Sun, Jiaoyan Chen 0001, Huajun Chen |
AAAI | 2 |
| 2020 | Logic-guided Semantic Representation Learning for Zero-Shot Relation ClassificationabstractRelation classification aims to extract semantic relations between entity pairs from the sentences.However, most existing methods can only identify seen relation classes that occurred during training.To recognize unseen relations at test time, we explore the problem of zero-shot relation classification.Previous work regards the problem as reading comprehension or textual entailment, which have to rely on artificial descriptive information to improve the understandability of relation types.Thus, rich semantic knowledge of the relation labels is ignored.In this paper, we propose a novel logic-guided semantic representation learning model for zero-shot relation classification.Our approach builds connections between seen and unseen relations via implicit and explicit semantic representations with knowledge graph embeddings and logic rules.Extensive experimental results demonstrate that our method can generalize to unseen relation types and achieve promising improvements. Juan Li 0010, Ruoxu Wang, Ningyu Zhang 0001, Wen Zhang 0015, Fan Yang 0087, Huajun Chen |
COLING | 3 |
| 2020 | Bridging Text and Knowledge with Multi-Prototype Embedding for Few-Shot Relational Triple ExtractionabstractCurrent supervised relational triple extraction approaches require huge amounts of labeled data and thus suffer from poor performance in few-shot settings.However, people can grasp new knowledge by learning a few instances.To this end, we take the first step to study the few-shot relational triple extraction, which has not been well understood.Unlike previous single-task few-shot problems, relational triple extraction is more challenging as the entities and relations have implicit correlations.In this paper, We propose a novel multi-prototype embedding network model to jointly extract the composition of relational triples, namely, entity pairs and corresponding relations.To be specific, we design a hybrid prototypical learning mechanism that bridges text and knowledge concerning both entities and relations.Thus, implicit correlations between entities and relations are injected.Additionally, we propose a prototype-aware regularization to learn more representative prototypes.Experimental results demonstrate that the proposed method can improve the performance of the few-shot triple extraction. Haiyang Yu 0003, Ningyu Zhang 0001, Shumin Deng, Hongbin Ye, Wei Zhang 0127, Huajun Chen |
COLING | 2 |
| 2020 | Meta-Learning with Dynamic-Memory-Based Prototypical Network for Few-Shot Event DetectionabstractEvent detection (ED), a sub-task of event extraction, involves identifying triggers and categorizing event mentions. Existing methods primarily rely upon supervised learning and require large-scale labeled event datasets which are unfortunately not readily available in many real-life applications. In this paper, we consider and reformulate the ED task with limited labeled data as a Few-Shot Learning problem. We propose a Dynamic-Memory-Based Prototypical Network (DMB-PN), which exploits Dynamic Memory Network (DMN) to not only learn better prototypes for event types, but also produce more robust sentence encodings for event mentions. Differing from vanilla prototypical networks simply computing event prototypes by averaging, which only consume event mentions once, our model is more robust and is capable of distilling contextual information from event mentions for multiple times due to the multi-hop mechanism of DMNs. The experiments show that DMB-PN not only deals with sample scarcity better than a series of baseline models but also performs more robustly when the variety of event types is relatively large and the instance quantity is extremely small. Shumin Deng, Ningyu Zhang 0001, Jiaojian Kang, Yichi Zhang 0010, Wei Zhang 0127, Huajun Chen |
WSDM | 2 |
| 2020 | Relation Adversarial Network for Low Resource Knowledge Graph CompletionabstractKnowledge Graph Completion (KGC) has been proposed to improve Knowledge Graphs by filling in missing connections via link prediction or relation extraction. One of the main difficulties for KGC is a low resource problem. Previous approaches assume sufficient training triples to learn versatile vectors for entities and relations, or a satisfactory number of labeled sentences to train a competent relation extraction model. However, low resource relations are very common in KGs, and those newly added relations often do not have many known samples for training. In this work, we aim at predicting new facts under a challenging setting where only limited training instances are available. We propose a general framework called Weighted Relation Adversarial Network, which utilizes an adversarial procedure to help adapt knowledge/features learned from high resource relations to different but related low resource relations. Specifically, the framework takes advantage of a relation discriminator to distinguish between samples from different relations, and help learn relation-invariant features more transferable from source relations to target relations. Experimental results show that the proposed approach outperforms previous methods regarding low resource settings for both link prediction and relation extraction. Ningyu Zhang 0001, Shumin Deng, Zhanlin Sun, Jiaoyan Chen 0001, Wei Zhang 0127, Huajun Chen |
WWW | 1 |
| 2018 | Attention-Based Capsule Network with Dynamic Routing for Relation ExtractionabstractA capsule is a group of neurons, whose activity vector represents the instantiation parameters of a specific type of entity.In this paper, we explore the capsule networks used for relation extraction in a multi-instance multilabel learning framework and propose a novel neural approach based on capsule networks with attention mechanisms.We evaluate our method with different benchmarks, and it is demonstrated that our method improves the precision of the predicted relations.Particularly, we show that capsule networks improve multiple entity pairs relation extraction 1 .* Corresponding author. 1 In this paper, multiple entity pairs relation extraction refers to multiple entity pairs in a single sentence and each pair of entities contains only one relation label. Ningyu Zhang 0001, Shumin Deng, Zhanling Sun, Xi Chen 0003, Wei Zhang 0127, Huajun Chen |
EMNLP | 1 |
| 2014 | HBaseSpatial: A Scalable Spatial Data Storage Based on HBaseabstractRecent years, the scale of spatial data is developing more and more huge and its storage has encountered a lot of problems. Traditional DBMS can efficiently handle some big spatial data. However, popular open source relational database systems are overwhelmed by the high insertion rates, querying requirements and terabytes of data that these systems can handle. On the other hand, key-value storage can effectively support large scale operations. To resolve the problems of big vector spatial data's storage and query, we bring forward HBase Spatial, a scalable spatial dada storage based on HBase. At first, we analyze the distributed storage model of HBase. Then, we design a distributed storage and index model. Finally, the advantages of our storage model and index algorithm are proven by experiments with both big sample sets and typical benchmarks on cluster compared with MongoDB and Mysql, which shows that our model can effectively enhance the query speed of big spatial data and provide a good solution for storage. Ningyu Zhang 0001, Guozhou Zheng, Huajun Chen, Jiaoyan Chen 0001, Xi Chen 0003 |
TrustCom | 1 |
| 2014 | Time-series processing of large scale remote sensing data with extreme learning machine
Jiaoyan Chen 0001, Guozhou Zheng, Cong Fang 0004, Ningyu Zhang 0001, Huajun Chen, Zhaohui Wu 0001 |
Neurocomputing | 4 |
| 2013 | OWL reasoning over big biomedical dataabstractRecently, the emerging accumulation of biomedical data on the Web (e.g. vast amounts of protein sequences, genes, gene products, drugs, diseases and chemical compounds, etc.) has shaped a big network of isolated professional knowledge. Embedded with domain knowledge from different disciplines all regarding to human biological systems, the decentralized data repositories are implicitly connected by human expert knowledge. Lots of biomedical data sources are published separately in the form of semantic ontologies represented by Web Ontology Language (OWL) syntax, which is naturally based on linked graphs. When we are faced with such massive, disparate and interlinked data, biomedical data analysis becomes a challenge. In this paper, we present a general OWL reasoning framework for the analysis of big biomedical data and implement a MapReduce-based property chain reasoning prototype system. OWL reasoning method is ideally suitable for problems involved complex semantic associations because it is able to infer logical consequences based on a set of asserted rules or axioms. MapReduce framework is used to solve the problem of scalability. In our experiment, we focus on the discovery of associations between Traditional Chinese Medicine (TCM) and Western Medicine (WM). The results show the system achieves high performance, accuracy and scalability. Xi Chen 0003, Huajun Chen, Ningyu Zhang 0001, Jiaoyan Chen 0001, Zhaohui Wu 0001 |
IEEE BigData | 3 |