Demo Data from SIGIR 2026 / reviewers in the wild / expert
Maosong Sun 0001
dblp:95/3291
· DBLP profile ↗
338ranked-venue papers
4as first author
171since 2021 · last 2026
0000-0002-6011-6115ORCID · conflict
Domains — the database's venue-derived domains; a paper can count in several
Artificial intelligence and machine learning · 300 · 4 first-author · 154 since 2021Graphics, computer vision, multimedia, augmented reality and games · 62 · 19 since 2021Databases, data management, data science and information retrieval · 23 · 10 since 2021Applied, interdisciplinary, general and emerging computing · 14 · 5 since 2021Human-computer interaction and ubiquitous computing · 2Systems, architecture and hardware · 1 · 1 since 2021
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
235 papers |
Language models and text generation · 29% Efficient and distributed learning · 12% Information extraction and text analysis · 11% | |
| Databases, data mining, and information retrieval
41 papers |
Information retrieval · 60% Knowledge graphs · 12% Web and social media mining · 11% | |
| Network and information security
9 papers |
Security and privacy of machine learning · 98% Systems and software security · 2% |
Topics — the 30 heaviest of 444, each with the papers that count most for it
| Topic | Weight | Papers | Last | Evidence papers |
|---|---|---|---|---|
Natural language and speech › Information extraction and text analysis
relation extraction |
6.3 | 18 | 2022 | Packed Levitated Marker for Entity and Relation Extraction · ACL (1) 2022 CodRED: A Cross-Document Relation Extraction Dataset for Acquiring Knowledge in the Wild · EMNLP (1) 2021 Denoising Relation Extraction from Document-level Distant Supervision · EMNLP (1) 2020 |
Computer vision › Vision and language › vision-language model
multimodal large language model |
5.4 | 7 | 2026 | LLaVA-UHD v2: Exploiting Hierarchical Vision Granularity in MLLMs via Inverse Semantic Pyramid · AAAI 2026 Benchmarking Retrieval-Augmented Generation in Multi-Modal Contexts · ACM Multimedia 2025 Looking Beyond Text: Reducing Language Bias in Large Vision-Language Models via Multimodal Dual-Attention and Soft-Image Guidance · EMNLP 2025 |
Natural language and speech › Language models and text generation
LLM agents |
4.5 | 5 | 2026 | RSMeM: Knowledge-Enhanced Memory Evolution for Remote Sensing Agents with Systematic Evaluation · ACL (1) 2026 Rational Decision-Making Agent with Learning Internal Utility Judgment · ICLR 2025 Scaling Large Language Model-based Multi-Agent Collaboration · ICLR 2025 |
Natural language and speech › Language models and text generation
retrieval-augmented generation |
4.4 | 5 | 2026 | Mixture-of-Retrieval Experts for Reasoning-Guided Multimodal Knowledge Exploitation · SIGIR 2026 CheckRLM: Effective Knowledge-Thought Coherence Checking in Retrieval-Augmented Reasoning · ACL (1) 2026 ParamMute: Suppressing Knowledge-Critical FFNs for Faithful Retrieval-Augmented Generation · NeurIPS 2025 |
Machine learning › Efficient and distributed learning
model compression |
4.1 | 6 | 2025 | Sparsing Law: Towards Large Language Models with Greater Activation Sparsity · ICML 2025 Cost-Optimal Grouped-Query Attention for Long-Context Modeling · EMNLP 2025 APB: Accelerating Distributed Long-Context Inference by Passing Compressed Context Blocks across GPUs · ACL (1) 2025 |
Natural language and speech › Machine translation
neural machine translation |
3.9 | 11 | 2023 | Continual Knowledge Distillation for Neural Machine Translation · ACL (1) 2023 Neural Machine Translation With Explicit Phrase Alignment · IEEE ACM Trans. Audio Speech Lang. Process. 2021 Reinforced Zero-Shot Cross-Lingual Neural Headline Generation · IEEE ACM Trans. Audio Speech Lang. Process. 2020 |
Natural language and speech › Language models and text generation
pre-trained language model |
3.5 | 7 | 2024 | Hyperbolic Pre-Trained Language Model · IEEE ACM Trans. Audio Speech Lang. Process. 2024 Plug-and-Play Knowledge Injection for Pre-trained Language Models · ACL (1) 2023 Moderate-fitting as a Natural Backdoor Defender for Pre-trained Language Models · NeurIPS 2022 |
Natural language and speech › Language models and text generation
instruction tuning |
3.4 | 5 | 2025 | From Unaligned to Aligned: Scaling Multilingual LLMs with Multi-Way Parallel Corpora · EMNLP 2025 Aligning Large Language Models to Follow Instructions and Hallucinate Less via Effective Data Filtering · ACL (1) 2025 ToolLLM: Facilitating Large Language Models to Master 16000+ Real-world APIs · ICLR 2024 |
Knowledge, reasoning and agents › Multi-agent systems › multi-agent collaboration
LLM-based multi-agent collaboration |
3.4 | 4 | 2025 | Multi-Agent Collaboration via Evolving Orchestration · NeurIPS 2025 Scaling Large Language Model-based Multi-Agent Collaboration · ICLR 2025 Internet of Agents: Weaving a Web of Heterogeneous Agents for Collaborative Intelligence · ICLR 2025 |
Machine learning › Efficient and distributed learning
parameter-efficient fine-tuning |
3.3 | 5 | 2024 | LoRA-Flow: Dynamic LoRA Fusion for Large Language Models in Generative Tasks · ACL (1) 2024 Exploring the Impact of Model Scaling on Parameter-Efficient Tuning · EMNLP 2023 Sparse Low-rank Adaptation of Pre-trained Language Models · EMNLP 2023 |
Natural language and speech › Language models and text generation
large language model |
3.2 | 5 | 2026 | IROTE: Human-like Traits Elicitation of Large Language Model via In-Context Self-Reflective Optimization · AAAI 2026 Sparsing Law: Towards Large Language Models with Greater Activation Sparsity · ICML 2025 Delta-CoMe: Training-Free Delta-Compression with Mixed-Precision for Large Language Models · NeurIPS 2024 |
Natural language and speech › Language models and text generation
alignment |
3.0 | 4 | 2024 | On the Essence and Prospect: An Investigation of Alignment Approaches for Big Models · IJCAI 2024 ULTRAFEEDBACK: Boosting Language Models with Scaled AI Feedback · ICML 2024 Controllable Preference Optimization: Toward Controllable Multi-Objective Alignment · EMNLP 2024 |
Information retrieval
retrieval-augmented generation |
2.9 | 4 | 2025 | Benchmarking Retrieval-Augmented Generation in Multi-Modal Contexts · ACM Multimedia 2025 VisRAG: Vision-based Retrieval-augmented Generation on Multi-modality Documents · ICLR 2025 RAGEval: Scenario Specific RAG Evaluation Dataset Generation Framework · ACL (1) 2025 |
Security and privacy of machine learning › adversarial attack
backdoor attack |
2.6 | 5 | 2022 | Moderate-fitting as a Natural Backdoor Defender for Pre-trained Language Models · NeurIPS 2022 Textual Backdoor Attacks Can Be More Harmful via Two Simple Tricks · EMNLP 2022 Mind the Style of Text! Adversarial and Backdoor Attacks Based on Text Style Transfer · EMNLP (1) 2021 |
Natural language and speech › Language models and text generation
hallucination mitigation |
2.6 | 3 | 2025 | Looking Beyond Text: Reducing Language Bias in Large Vision-Language Models via Multimodal Dual-Attention and Soft-Image Guidance · EMNLP 2025 RLAIF-V: Open-Source AI Feedback Leads to Super GPT-4V Trustworthiness · CVPR 2025 Aligning Large Language Models to Follow Instructions and Hallucinate Less via Effective Data Filtering · ACL (1) 2025 |
Natural language and speech › Language models and text generation
multilingual language models |
2.5 | 3 | 2025 | DCAD-2000: A Multilingual Dataset across 2000+ Languages with Data Cleaning as Anomaly Detection · NeurIPS 2025 From Unaligned to Aligned: Scaling Multilingual LLMs with Multi-Way Parallel Corpora · EMNLP 2025 Large Multilingual Models Pivot Zero-Shot Multimodal Learning across Languages · ICLR 2024 |
Machine learning › Trustworthy machine learning
robustness |
2.5 | 5 | 2025 | Looking Beyond Text: Reducing Language Bias in Large Vision-Language Models via Multimodal Dual-Attention and Soft-Image Guidance · EMNLP 2025 Revisiting Out-of-distribution Robustness in NLP: Benchmarks, Analysis, and LLMs Evaluations · NeurIPS 2023 Beat LLMs at Their Own Game: Zero-Shot LLM-Generated Text Detection via Querying ChatGPT · EMNLP 2023 |
Computer vision › Vision and language
vision-language pretraining |
2.5 | 4 | 2023 | Weakly Supervised Vision-and-Language Pre-training with Relative Representations · ACL (1) 2023 Visually Grounded Commonsense Knowledge Acquisition · AAAI 2023 PEVL: Position-enhanced Pre-training and Prompt Tuning for Vision-language Models · EMNLP 2022 |
Natural language and speech › Question answering and dialogue systems
open-domain question answering |
2.5 | 4 | 2026 | Mixture-of-Retrieval Experts for Reasoning-Guided Multimodal Knowledge Exploitation · SIGIR 2026 FastFiD: Improve Inference Efficiency of Open Domain Question Answering via Sentence Selection · ACL (1) 2024 XQA: A Cross-lingual Open-domain Question Answering Dataset · ACL (1) 2019 |
Natural language and speech › Language models and text generation
large language model evaluation |
2.4 | 4 | 2026 | Can Large Language Models Analyze Graphs like Professionals? A Benchmark, Datasets and Models · NeurIPS 2024 Revisiting Out-of-distribution Robustness in NLP: Benchmarks, Analysis, and LLMs Evaluations · NeurIPS 2023 C-Eval: A Multi-Level Multi-Discipline Chinese Evaluation Suite for Foundation Models · NeurIPS 2023 |
Natural language and speech › Language models and text generation
preference optimization |
2.2 | 4 | 2026 | Chunks as Arms: Multi-Armed Bandit-Guided Sampling for Long-Context LLM Preference Optimization · ACL (1) 2026 Controllable Preference Optimization: Toward Controllable Multi-Objective Alignment · EMNLP 2024 NotaGen: Advancing Musicality in Symbolic Music Generation with Large Language Model Training Paradigms · IJCAI 2025 |
Security and privacy of machine learning › adversarial attack › backdoor attack
textual backdoor attack |
2.1 | 4 | 2022 | Textual Backdoor Attacks Can Be More Harmful via Two Simple Tricks · EMNLP 2022 Mind the Style of Text! Adversarial and Backdoor Attacks Based on Text Style Transfer · EMNLP (1) 2021 Turn the Combination Lock: Learnable Textual Backdoor Attacks via Word Substitution · ACL/IJCNLP (1) 2021 |
Machine learning › Reinforcement learning › reward learning
reward modeling |
2.0 | 3 | 2025 | Advancing LLM Reasoning Generalists with Preference Trees · ICLR 2025 Proactive Agent: Shifting LLM Agents from Reactive Responses to Active Assistance · ICLR 2025 AgentRM: Enhancing Agent Generalization with Reward Modeling · ACL (1) 2025 |
Machine learning › Efficient and distributed learning
inference efficiency |
1.9 | 3 | 2025 | FR-Spec: Accelerating Large-Vocabulary Language Models via Frequency-Ranked Speculative Sampling · ACL (1) 2025 FastFiD: Improve Inference Efficiency of Open Domain Question Answering via Sentence Selection · ACL (1) 2024 InfLLM: Training-Free Long-Context Extrapolation for LLMs with an Efficient Context Memory · NeurIPS 2024 |
Machine learning › Trustworthy machine learning
hallucination |
1.8 | 2 | 2026 | Teaching Large Language Models to Maintain Contextual Faithfulness via Synthetic Tasks and Reinforcement Learning · AAAI 2026 RLHF-V: Towards Trustworthy MLLMs via Behavior Alignment from Fine-Grained Correctional Human Feedback · CVPR 2024 |
Machine learning › Efficient and distributed learning
model merging |
1.7 | 2 | 2025 | AdaMMS: Model Merging for Heterogeneous Multimodal Large Language Models with Unsupervised Coefficient Optimization · CVPR 2025 Fusing Highly Specialized Language Models for Comprehensive Expertise · ACL (1) 2025 |
Information retrieval › retrieval-augmented generation
multimodal retrieval-augmented generation |
1.7 | 2 | 2025 | Benchmarking Retrieval-Augmented Generation in Multi-Modal Contexts · ACM Multimedia 2025 VisRAG: Vision-based Retrieval-augmented Generation on Multi-modality Documents · ICLR 2025 |
Machine learning › Efficient and distributed learning › model compression
knowledge distillation |
1.7 | 2 | 2026 | Long-Chain Reasoning Distillation via Adaptive Prefix Alignment · ACL (1) 2026 Continual Knowledge Distillation for Neural Machine Translation · ACL (1) 2023 |
Security and privacy of machine learning › adversarial attack › backdoor attack
backdoor defense |
1.6 | 3 | 2022 | Moderate-fitting as a Natural Backdoor Defender for Pre-trained Language Models · NeurIPS 2022 A Unified Evaluation of Textual Backdoor Learning: Frameworks and Benchmarks · NeurIPS 2022 ONION: A Simple and Effective Defense Against Textual Backdoor Attacks · EMNLP (1) 2021 |
Machine learning › Efficient and distributed learning › model compression › sparsity
activation sparsity |
1.6 | 2 | 2025 | Sparsing Law: Towards Large Language Models with Greater Activation Sparsity · ICML 2025 Exploring the Benefit of Activation Sparsity in Pre-training · ICML 2024 |
Methods — techniques the papers use, named apart from their topics
fine-tuning · 8.3reinforcement learning · 8.2instruction tuning · 6.8large language model · 6.0large language model prompting · 3.4prompt tuning · 3.2contrastive learning · 2.6neural network · 2.3knowledge distillation · 2.0mixture of experts · 1.8large language model agent · 1.5data parallelism · 1.3recurrent neural network · 1.2self-reflection · 1.0paper lineage analysis · 1.0memory evolution · 1.0knowledge enhancement · 1.0information-theoretic objective · 1.0
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2026 | IROTE: Human-like Traits Elicitation of Large Language Model via In-Context Self-Reflective OptimizationabstractTrained on various human-authored corpora, Large Language Models (LLMs) have demonstrated a certain capability of reflecting specific human-like traits (e.g., personality or values) by prompting, benefiting applications like personalized LLMs and social simulations. However, existing methods suffer from the superficial elicitation problem: LLMs can only be steered to mimic shallow and unstable stylistic patterns, failing to embody the desired traits precisely and consistently across diverse tasks like humans. To address this challenge, we propose IROTE, a novel in-context method for stable and transferable trait elicitation. Drawing on psychological theories suggesting that traits are formed through identity-related reflection, our method automatically generates and optimizes a textual self-reflection within prompts, which comprises self-perceived experience, to stimulate LLMs' trait-driven behavior. The optimization is performed by iteratively maximizing an information-theoretic objective that enhances the connections between LLMs' behavior and the target trait, while reducing noisy redundancy in reflection without any fine-tuning, leading to evocative and compact trait reflection. Extensive experiments across three human trait systems manifest that one single IROTE-generated self-reflection can induce LLMs' stable impersonation of the target trait across diverse downstream tasks beyond simple questionnaire answering, consistently outperforming existing strong baselines. Yuzhuo Bai, Shitong Duan, Muhua Huang, Jing Yao 0003, Zhenghao Liu 0001, Peng Zhang 0060, Tun Lu, Xiaoyuan Yi, Maosong Sun 0001, Xing Xie 0001 |
AAAI | 9 |
| 2026 | Teaching Large Language Models to Maintain Contextual Faithfulness via Synthetic Tasks and Reinforcement LearningabstractTeaching large language models (LLMs) to be faithful in the provided context is crucial for building reliable information-seeking systems. Therefore, we propose a systematic framework, CANOE, to reduce faithfulness hallucinations of LLMs across different downstream tasks without human annotations. Specifically, we first synthesize short-form question-answering (QA) data with four diverse tasks to construct high-quality and easily verifiable training data without human annotation. Also, we propose Dual-GRPO, a rule-based reinforcement learning method that includes three tailored rule-based rewards derived from synthesized short-form QA data, while simultaneously optimizing both short-form and long-form response generation. Notably, Dual-GRPO eliminates the need to manually label preference data to train reward models and avoids over-optimizing short-form generation when relying only on the synthesized short-form QA data. Experimental results show that CANOE greatly improves the faithfulness of LLMs across 11 different tasks, even outperforming the most advanced LLMs, e.g., GPT-4o and OpenAI o1. Shuzheng Si, Haozhe Zhao, Yuzhuo Bai, Zhitong Wang, Bofei Gao, Kangyang Luo, Wenhao Li 0003, Yufei Huang 0008, Gang Chen 0039, Fanchao Qi, Minjia Zhang, Baobao Chang, Maosong Sun 0001 |
AAAI | 14 |
| 2026 | LLaVA-UHD v2: Exploiting Hierarchical Vision Granularity in MLLMs via Inverse Semantic PyramidabstractVision transformers (ViTs) are widely employed in multimodal large language models (MLLMs) for visual encoding. However, they exhibit inferior performance on tasks regarding fine-grained visual perception. We attribute this to the inner limitations of ViTs in capturing diverse visual semantic levels. To address this, we present Hierarchical window (Hiwin) transformer as a plug-and-play solution for MLLMs, centered around our inverse semantic pyramid (ISP). Hiwin transformer comprises two key modules: (i) a visual detail injection module, which progressively injects low-level visual details into high-level language-aligned semantics features, thereby constructing an ISP, and (ii) a hierarchical window attention module, which leverages cross-scale windows to condense multi-level semantics from the ISP. Notably, our design achieves an average boost of 3.7% across 14 benchmarks compared with the baseline method, 9.3% on DocVQA for instance. Zonghao Guo, Xuesong Yang, Chi Chen 0005, Yuan Yao 0013, Tat-Seng Chua, Maosong Sun 0001 |
AAAI | 11 |
| 2026 | Musical Score Understanding Benchmark: Evaluating Large Language Models' Comprehension of Complete Musical ScoresabstractCongren Dai, Yue Yang, Krinos Li, Huichi Zhou, Shijie Liang, Zhang Bo, Enyang Liu, Ge Jin, Hongran An, Haosen Zhang, Peiyuan Jing, KinHei Lee, Zhenxuan Zhang, Xiaobing Li, Maosong Sun. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Congren Dai, Krinos Li, Huichi Zhou, Shijie Liang, Zhang Bo, Enyang Liu, Hongran An, Haosen Zhang, Peiyuan Jing, Kinhei Lee, Zhenxuan Zhang, Maosong Sun 0001 |
ACL (1) | 15 |
| 2026 | Chunks as Arms: Multi-Armed Bandit-Guided Sampling for Long-Context LLM Preference OptimizationabstractShaohua Duan, Pengcheng Huang, Xinze Li, Zhenghao Liu, Xiaoyuan Yi, Yukun Yan, Shuo Wang, Yu Gu, Ge Yu, Maosong Sun. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Shaohua Duan, Pengcheng Huang 0004, Zhenghao Liu 0001, Xiaoyuan Yi, Yukun Yan, Shuo Wang 0013, Yu Gu 0002, Ge Yu 0001, Maosong Sun 0001 |
ACL (1) | 10 |
| 2026 | Empirical Analysis of Decoding Biases in Masked Diffusion ModelsabstractPengcheng Huang, Tianming Liu, Zhenghao Liu, Yukun Yan, Shuo Wang, Tong Xiao, Zulong Chen, Maosong Sun. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Pengcheng Huang 0004, Zhenghao Liu 0001, Yukun Yan, Shuo Wang 0013, Tong Xiao 0001, Zulong Chen, Maosong Sun 0001 |
ACL (1) | 8 |
| 2026 | UNIKIE-BENCH: Benchmarking Large Multimodal Models for Key Information Extraction in Visual DocumentsabstractYifan Ji, Zhipeng Xu, Zhenghao Liu, Zulong Chen, Qian Zhang, Zhibo Yang, Junyang Lin, Yu Gu, Ge Yu, Maosong Sun. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Yifan Ji, Zhenghao Liu 0001, Zulong Chen, Zhibo Yang 0003, Junyang Lin, Yu Gu 0002, Ge Yu 0001, Maosong Sun 0001 |
ACL (1) | 10 |
| 2026 | Long-Chain Reasoning Distillation via Adaptive Prefix AlignmentabstractZhenghao Liu, Zhuoyang Wu, Xinze Li, Yukun Yan, Shuo Wang, Zulong Chen, Yu Gu, Ge Yu, Maosong Sun. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Zhenghao Liu 0001, Zhuoyang Wu, Yukun Yan, Shuo Wang 0013, Zulong Chen, Yu Gu 0002, Ge Yu 0001, Maosong Sun 0001 |
ACL (1) | 9 |
| 2026 | ImCoref-CeS: An Improved Lightweight Pipeline for Coreference Resolution with LLM-based Checker-Splitter RefinementabstractKangyang Luo, Yuzhuo Bai, Shuzheng Si, Cheng Gao, Zhitong Wang, Yingli Shen, Wenhao Li, Zhu Liu, Yufeng Han, Jiayi Wu, Cunliang Kong, Maosong Sun. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Kangyang Luo, Yuzhuo Bai, Shuzheng Si, Zhitong Wang, Yingli Shen, Wenhao Li 0003, Zhu Liu 0005, Yufeng Han, Cunliang Kong, Maosong Sun 0001 |
ACL (1) | 12 |
| 2026 | A Goal Without a Plan Is Just a Wish: Efficient and Effective Global Planner Training for Long-Horizon Agent TaskabstractShuzheng Si, Haozhe Zhao, Kangyang Luo, Gang Chen, Fanchao Qi, Minjia Zhang, Baobao Chang, Maosong Sun. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Shuzheng Si, Haozhe Zhao, Kangyang Luo, Gang Chen 0039, Fanchao Qi, Minjia Zhang, Baobao Chang, Maosong Sun 0001 |
ACL (1) | 8 |
| 2026 | RSMeM: Knowledge-Enhanced Memory Evolution for Remote Sensing Agents with Systematic EvaluationabstractBingxian Wu, Yu Zhang, Zonghao Guo, Tang Liu, Chen Qian, Yuxiang Lu, Xingbo Du, Yanghao Li, Yidan Zhang, Chi Chen, Ling Yao, Chenghu Zhou, Maosong Sun. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Bingxian Wu, Yu Zhang 0186, Zonghao Guo, Xingbo Du, Yanghao Li, Chi Chen 0005, Ling Yao, Chenghu Zhou, Maosong Sun 0001 |
ACL (1) | 13 |
| 2026 | CheckRLM: Effective Knowledge-Thought Coherence Checking in Retrieval-Augmented ReasoningabstractDingling Xu, Ruobing Wang, Qingfei Zhao, Yukun Yan, Zhichun Wang, Daren Zha, Shi Yu, Zhenghao Liu, Shuo Wang, Xu Han, Maosong Sun. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Dingling Xu, Qingfei Zhao, Yukun Yan, Zhichun Wang, Daren Zha, Shi Yu 0001, Zhenghao Liu 0001, Shuo Wang 0013, Maosong Sun 0001 |
ACL (1) | 11 |
| 2026 | AutoReproduce: Automatic AI Experiment Reproduction with Paper LineageabstractXuanle Zhao, Zilin Sang, Yuxuan Li, Qi Shi, Weilun Zhao, Shuo Wang, Duzhen Zhang, Xu Han, Zhiyuan Liu, Maosong Sun. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Xuanle Zhao, Zilin Sang, Qi Shi 0002, Wei-Lun Zhao, Shuo Wang 0013, Duzhen Zhang, Xu Han 0007, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 10 |
| 2026 | Mixture-of-Retrieval Experts for Reasoning-Guided Multimodal Knowledge ExploitationabstractMultimodal Retrieval-Augmented Generation (MRAG) has shown promise in mitigating hallucinations in Multimodal Large Language Models (MLLMs) by incorporating external knowledge. However, existing methods typically adhere to rigid retrieval paradigms by mimicking fixed retrieval trajectories and thus fail to fully exploit the knowledge of different retrieval experts through dynamic interaction based on the model's knowledge needs or evolving reasoning states. To overcome this limitation, we introduce Mixture-of-Retrieval Experts (MoRE), a novel framework that enables MLLMs to collaboratively interact with diverse retrieval experts for more effective knowledge exploitation. Specifically, MoRE learns to dynamically determine which expert to engage with, conditioned on the evolving reasoning state. To effectively train this capability, we propose Stepwise Group Relative Policy Optimization (Step-GRPO), which goes beyond sparse outcome-based supervision by encouraging MLLMs to interact with multiple retrieval experts and synthesize fine-grained rewards, thereby teaching the MLLM to fully coordinate all experts when answering a given query. Experimental results on diverse open-domain QA benchmarks demonstrate the effectiveness of MoRE, achieving average performance gains of over 7% compared to competitive baselines. Notably, MoRE exhibits strong adaptability by dynamically coordinating heterogeneous experts to precisely locate relevant information, validating its capability for robust, reasoning-driven expert collaboration. All codes and data are released on https://github.com/OpenBMB/MoRE. Zhenghao Liu 0001, Yishan Li, Yukun Yan, Shuo Wang 0013, Yu Gu 0002, Minghe Yu 0001, Ge Yu 0001, Maosong Sun 0001 |
SIGIR | 10 |
| 2026 | From MOOC to MAIC: Reimagine Online Teaching and Learning Through LLM-Driven Agents
Jifan Yu, Daniel Zhang-Li, Zhe-Yuan Zhang, Yu-Cheng Wang, Haoxuan Li 0003, Joy Lim Jia Yin, Zhan-Xin Hao, Shang-Qing Tu, Lu Zhang 0096, Xu-Sheng Dai, Jian-Xiao Jiang, Bing-Lin Liu, Xin Cong, Bin Xu 0001, Lei Hou 0001, Man-Li Li, Juan-Zi Li, Hui-Qin Liu, Yu Zhang 0186, Zhiyuan Liu 0001, Maosong Sun 0001 |
J. Comput. Sci. Technol. | 24 |
| 2025 | Enhancing Open-Domain Task-Solving Capability of LLMs via Autonomous Tool Integration from GitHubabstractBohan Lyu, Xin Cong, Heyang Yu, Pan Yang, Cheng Qian, Zihe Wang, Yujia Qin, Yining Ye, Yaxi Lu, Chen Qian, Zhong Zhang, Yukun Yan, Yankai Lin, Zhiyuan Liu, Maosong Sun. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025. Bohan Lyu 0001, Xin Cong, Heyang Yu, Pan Yang 0022, Cheng Qian 0008, Yujia Qin, Yining Ye, Yaxi Lu, Zhong Zhang 0004, Yukun Yan, Yankai Lin 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 15 |
| 2025 | Fusing Highly Specialized Language Models for Comprehensive ExpertiseabstractNing Ding, Yulin Chen, Ganqu Cui, Xingtai Lv, Weilin Zhao, Kaiyan Zhang, Ruobing Xie, Bowen Zhou, Zhiyuan Liu, Maosong Sun. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025. Ning Ding 0002, Yulin Chen 0001, Ganqu Cui, Xingtai Lv, Weilin Zhao, Ruobing Xie, Bowen Zhou 0002, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 10 |
| 2025 | GUICourse: From General Vision Language Model to Versatile GUI AgentabstractUtilizing Graphic User Interfaces (GUIs) for human-computer interaction is essential for accessing various digital tools. Recent advancements in Vision Language Models (VLMs) reveal significant potential for developing versatile agents that assist humans in navigating GUIs. However, current VLMs face challenges related to fundamental abilities, such as OCR and grounding, as well as a lack of knowledge about GUI elements functionalities and control methods. These limitations hinder their effectiveness as practical GUI agents. To address these challenges, we introduce GUICourse, a series of datasets for training visual-based GUI agents using general VLMs. First, we enhance the OCR and grounding capabilities of VLMs using the GUIEnv dataset. Next, we enrich the GUI knowledge of VLMs using the GUIAct and GUIChat datasets. Our experiments demonstrate that even a small-sized GUI agent (with 3.1 billion parameters) performs effectively on both single-step and multi-step GUI tasks. We further finetune our GUI agents on other GUI tasks with different action spaces (AITW and Mind2Web), and the results show that our agents are better than their baseline VLMs. Additionally, we analyze the impact of OCR and grounding capabilities through an ablation study, revealing a positive correlation with GUI navigation ability. Wentong Chen, Junbo Cui, Jinyi Hu, Yujia Qin, Junjie Fang, Chongyi Wang, Guirong Chen, Yupeng Huo, Yuan Yao 0013, Yankai Lin 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 14 |
| 2025 | APB: Accelerating Distributed Long-Context Inference by Passing Compressed Context Blocks across GPUsabstractWhile long-context inference is crucial for advancing large language model (LLM) applications, its prefill speed remains a significant bottleneck. Current approaches, including sequence parallelism strategies and compute reduction through approximate attention mechanisms, still fall short of delivering optimal inference efficiency. This hinders scaling the inputs to longer sequences and processing long-context queries in a timely manner. To address this, we introduce APB, an efficient long-context inference framework that leverages multi-host approximate attention to enhance prefill speed by reducing compute and enhancing parallelism simultaneously. APB introduces a communication mechanism for essential key-value pairs within a sequence parallelism framework, enabling a faster inference speed while maintaining task performance. We implement APB by incorporating a tailored FlashAttn kernel alongside optimized distribution strategies, supporting diverse models and parallelism configurations. APB achieves speedups of up to 9.2\times, 4.2\times, and 1.6\times compared with FlashAttn, RingAttn, and StarAttn, respectively, without any observable task performance degradation. Yuxiang Huang 0001, Mingye Li, Xu Han 0007, Chaojun Xiao, Weilin Zhao, Sun Ao, Hao Zhou 0012, Jie Zhou 0016, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 10 |
| 2025 | Learning to Generate Structured Output with Schema Reinforcement LearningabstractYaxi Lu, Haolun Li, Xin Cong, Zhong Zhang, Yesai Wu, Yankai Lin, Zhiyuan Liu, Fangming Liu, Maosong Sun. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025. Yaxi Lu, Haolun Li 0003, Xin Cong, Zhong Zhang 0004, Yesai Wu, Yankai Lin 0001, Zhiyuan Liu 0001, Fangming Liu, Maosong Sun 0001 |
ACL (1) | 9 |
| 2025 | Aligning Large Language Models to Follow Instructions and Hallucinate Less via Effective Data FilteringabstractShuzheng Si, Haozhe Zhao, Gang Chen, Cheng Gao, Yuzhuo Bai, Zhitong Wang, Kaikai An, Kangyang Luo, Chen Qian, Fanchao Qi, Baobao Chang, Maosong Sun. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025. Shuzheng Si, Haozhe Zhao, Gang Chen 0039, Yuzhuo Bai, Zhitong Wang, Kaikai An, Kangyang Luo, Fanchao Qi, Baobao Chang, Maosong Sun 0001 |
ACL (1) | 12 |
| 2025 | AgentRM: Enhancing Agent Generalization with Reward ModelingabstractYu Xia, Jingru Fan, Weize Chen, Siyu Yan, Xin Cong, Zhong Zhang, Yaxi Lu, Yankai Lin, Zhiyuan Liu, Maosong Sun. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025. Jingru Fan, Weize Chen, Xin Cong, Zhong Zhang 0004, Yaxi Lu, Yankai Lin 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 10 |
| 2025 | ChartCoder: Advancing Multimodal Large Language Model for Chart-to-Code GenerationabstractMultimodal Large Language Models (MLLMs) have demonstrated remarkable capabilities in chart understanding tasks.However, interpreting charts with textual descriptions often leads to information loss, as it fails to fully capture the dense information embedded in charts.In contrast, parsing charts into code provides lossless representations that can effectively contain all critical details.Although existing open-source MLLMs have achieved success in chart understanding tasks, they still face two major challenges when applied to chart-to-code tasks: (1) Low executability and poor restoration of chart details in the generated code and (2) Lack of large-scale and diverse training data.To address these challenges, we propose ChartCoder, the first dedicated chart-to-code MLLM, which leverages Code LLMs as the language backbone to enhance the executability of the generated code.Furthermore, we introduce Chart2Code-160k, the first large-scale and diverse dataset for chartto-code generation, and propose the Snippetof-Thought (SoT) method, which transforms direct chart-to-code generation data into stepby-step generation.Experiments demonstrate that ChartCoder, with only 7B parameters, surpasses existing open-source MLLMs on chartto-code benchmarks, achieving superior chart restoration and code excitability.Our code is available at https://github.com/thunlp/ ChartCoder.89 seed code with 27 chart types Available functions and parameters Xuanle Zhao, Xianzhen Luo, Qi Shi 0002, Chi Chen 0005, Shuo Wang 0013, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 7 |
| 2025 | FR-Spec: Accelerating Large-Vocabulary Language Models via Frequency-Ranked Speculative SamplingabstractWeilin Zhao, Tengyu Pan, Xu Han, Yudi Zhang, Sun Ao, Yuxiang Huang, Kaihuo Zhang, Weilun Zhao, Yuxuan Li, Jie Zhou, Hao Zhou, Jianyong Wang, Maosong Sun, Zhiyuan Liu. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025. Weilin Zhao, Tengyu Pan, Xu Han 0007, Sun Ao, Yuxiang Huang 0001, Kaihuo Zhang, Wei-Lun Zhao, Jie Zhou 0016, Hao Zhou 0012, Jianyong Wang 0001, Maosong Sun 0001, Zhiyuan Liu 0001 |
ACL (1) | 13 |
| 2025 | LLM×MapReduce: Simplified Long-Sequence Processing using Large Language ModelsabstractZihan Zhou, Chong Li, Xinyi Chen, Shuo Wang, Yu Chao, Zhili Li, Haoyu Wang, Qi Shi, Zhixing Tan, Xu Han, Xiaodong Shi, Zhiyuan Liu, Maosong Sun. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025. Shuo Wang 0013, Yu Chao, Zhili Li, Qi Shi 0002, Zhixing Tan, Xu Han 0007, Xiaodong Shi, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 13 |
| 2025 | RAGEval: Scenario Specific RAG Evaluation Dataset Generation FrameworkabstractKunlun Zhu, Yifan Luo, Dingling Xu, Yukun Yan, Zhenghao Liu, Shi Yu, Ruobing Wang, Shuo Wang, Yishan Li, Nan Zhang, Xu Han, Zhiyuan Liu, Maosong Sun. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025. Kunlun Zhu, Dingling Xu, Yukun Yan, Zhenghao Liu 0001, Shi Yu 0001, Shuo Wang 0013, Yishan Li, Xu Han 0007, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 13 |
| 2025 | Multi-Modal Multi-Granularity Tokenizer for Chu Bamboo SlipsabstractThis study presents a multi-modal multi-granularity tokenizer specifically designed for analyzing ancient Chinese scripts, focusing on the Chu bamboo slip (CBS) script used during the Spring and Autumn and Warring States period (771-256 BCE) in Ancient China. Considering the complex hierarchical structure of ancient Chinese scripts, where a single character may be a combination of multiple sub-characters, our tokenizer first adopts character detection to locate character boundaries. Then it conducts character recognition at both the character and sub-character levels. Moreover, to support the academic community, we assembled the first large-scale dataset of CBSs with over 100K annotated character image scans. On the part-of-speech tagging task built on our dataset, using our tokenizer gives a 5.5% relative improvement in F1-score compared to mainstream sub-word tokenizers. Our work not only aids in further investigations of the specific script but also has the potential to advance research on other forms of ancient Chinese scripts. Yingfa Chen, Chenlong Hu, Shi Yu 0001, Xu Han 0007, Zhiyuan Liu 0001, Maosong Sun 0001 |
COLING | 8 |
| 2025 | ProSparse: Introducing and Enhancing Intrinsic Activation Sparsity within Large Language ModelsabstractActivation sparsity refers to the existence of considerable weakly-contributed elements among activation outputs, serving as a promising paradigm for accelerating model inference. Nevertheless, most large language models (LLMs) adopt activation functions without intrinsic activation sparsity (e.g., GELU and Swish). Some recent efforts have explored introducing ReLU or its variants as the substitutive activation function to pursue activation sparsity and acceleration, but few can simultaneously obtain high activation sparsity and comparable model performance. This paper introduces a simple and effective method named “ProSparse” to sparsify LLMs while achieving both targets. Specifically, after introducing ReLU activation, ProSparse adopts progressive sparsity regularization with a factor smoothly increasing for multiple stages. This can enhance activation sparsity and mitigate performance degradation by avoiding radical shifts in activation distributions. With ProSparse, we obtain high sparsity of 89.32% for LLaMA2-7B, 88.80% for LLaMA2-13B, and 87.89% for end-size MiniCPM-1B, respectively, with comparable performance to their original Swish-activated versions. These present the most sparsely activated models among open-source LLaMA versions and competitive end-size models. Inference acceleration experiments further demonstrate the significant practical acceleration potential of LLMs with higher activation sparsity, obtaining up to 4.52x inference speedup. Xu Han 0007, Zhengyan Zhang, Shengding Hu, Xiyu Shi, Kuai Li, Zhiyuan Liu 0001, Guangli Li, Maosong Sun 0001 |
COLING | 11 |
| 2025 | RLAIF-V: Open-Source AI Feedback Leads to Super GPT-4V TrustworthinessabstractTraditional feedback learning for hallucination reduction relies on labor-intensive manual labeling or expensive proprietary models. This leaves the community without foundational knowledge about how to build high-quality feedback with open-source MLLMs. In this work, we introduce RLAIF-V, a novel framework that aligns MLLMs in a fully open-source paradigm. RLAIF-V maximally explores open-source MLLMs from two perspectives, including high-quality feedback data generation for preference learning and self-feedback guidance for inference-time scaling. Extensive experiments on six benchmarks in both automatic and human evaluation show that RLAIF-V substantially enhances the trustworthiness of models at both preference learning and inference time. RLAIF-V 7B reduces object hallucination by 80.7% and overall hallucination by 33.7%. Remarkably, RLAIF-V 12B further reveals the self-alignment potential of open-source MLLMs, where the model can learn from feedback of itself to achieve super GPT-4V trustworthiness. Tianyu Yu 0002, Haoye Zhang, Qixin Xu, Yuan Yao 0013, Xiaoman Lu, Ganqu Cui, Yunkai Dang, Taiwen He, Bo Zheng 0007, Zhiyuan Liu 0001, Tat-Seng Chua, Maosong Sun 0001 |
CVPR | 16 |
| 2025 | XLRS-Bench: Could Your Multimodal LLMs Understand Extremely Large Ultra-High-Resolution Remote Sensing Imagery?abstractThe astonishing breakthrough of multimodal large language models (MLLMs) has necessitated new benchmarks to quantitatively assess their capabilities, reveal their limitations, and indicate future research directions. However, this is challenging in the context of remote sensing (RS), since the imagery features ultra-high resolution that incorporates extremely complex semantic relationships. Existing benchmarks usually adopt notably smaller image sizes than real-world RS scenarios, suffer from limited annotation quality, and consider insufficient dimensions of evaluation. To address these issues, we present XLRS-Bench: a comprehensive benchmark for evaluating the perception and reasoning capabilities of MLLMs in ultra-high-resolution RS scenarios. XLRS-Bench boasts the largest average image size (8500×8500) observed thus far, with all evaluation samples meticulously annotated manually, assisted by a novel semi-automatic captioner on ultra-high-resolution RS images. On top of the XLRS-Bench, 16 sub-tasks are defined to evaluate MLLMs’ 10 kinds of perceptual capabilities and 6 kinds of reasoning capabilities, with a primary emphasis on advanced cognitive processes that facilitate real-world decision-making and the capture of spatiotemporal changes. The results of both general and RS-focused MLLMs on XLRS-Bench indicate that further efforts are needed for real-world RS applications. We have open-sourced XLRS-Bench to support further research in developing more powerful MLLMs for remote sensing. Fengxiang Wang 0004, Hongzhen Wang, Zonghao Guo, Di Wang 0023, Yulin Wang 0002, Mingshuo Chen, Long Lan, Wenjing Yang 0002, Jing Zhang 0037, Zhiyuan Liu 0001, Maosong Sun 0001 |
CVPR | 12 |
| 2025 | AdaMMS: Model Merging for Heterogeneous Multimodal Large Language Models with Unsupervised Coefficient OptimizationabstractRecently, model merging methods have demonstrated powerful strengths in combining abilities on various tasks from multiple Large Language Models (LLMs). While previous model merging methods mainly focus on merging homogeneous models with identical architecture, they meet challenges when dealing with Multimodal Large Language Models (MLLMs) with inherent heterogeneous property, including differences in model architecture and the asymmetry in the parameter space. In this work, we propose AdaMMS1, a novel model merging method tailored for heterogeneous MLLMs. Our method tackles the challenges in three steps: mapping, merging and searching. Specifically, we first design mapping function between models to apply model merging on MLLMs with different architecture. Then we apply linear interpolation on model weights to actively adapt the asymmetry in the heterogeneous MLLMs. Finally in the hyper-parameter searching step, we propose an unsupervised hyper-parameter selection method for model merging. As the first model merging method capable of merging heterogeneous MLLMs without labeled data, extensive experiments on various model combinations demonstrated that AdaMMS outperforms previous model merging methods on various vision-language benchmarks.2 Yiyang Du, Xiaochen Wang 0002, Chi Chen 0005, Jiabo Ye, Peng Li 0030, Ming Yan 0008, Ji Zhang 0011, Fei Huang 0002, Zhifang Sui, Maosong Sun 0001, Yang Liu 0005 |
CVPR | 11 |
| 2025 | Cost-Optimal Grouped-Query Attention for Long-Context ModelingabstractGrouped-Query Attention (GQA) is a widely adopted strategy for reducing the computational cost of attention layers in large language models (LLMs).However, current GQA configurations are often suboptimal because they overlook how context length influences inference cost.Since inference cost grows with context length, the most cost-efficient GQA configuration should vary accordingly.In this work, we analyze the relationship among context length, model size, GQA configuration, and model loss, and introduce two innovations:(1) we decouple the total head size from the hidden size, enabling more flexible control over attention FLOPs; and (2) we jointly optimize the model size and the GQA configuration to arrive at a better allocation of inference resources between attention layers and other components.Our analysis reveals that commonly used GQA configurations are highly suboptimal for longcontext scenarios.Moreover, we propose a recipe for deriving cost-optimal GQA configurations.Our results show that for long-context scenarios, one should use fewer attention heads while scaling up the model size.Configurations selected by our recipe can reduce both memory usage and FLOPs by more than 50% compared to Llama-3's GQA, with no degradation in model capabilities.Our findings offer valuable insights for designing efficient longcontext LLMs. 1 Memory (GB)-57.8% -50.8% Yingfa Chen, Zhen Leng Thai, Xu Han 0007, Zhiyuan Liu 0001, Maosong Sun 0001 |
EMNLP | 8 |
| 2025 | On LLM-Based Scientific Inductive Reasoning Beyond EquationsabstractBrian S. Lin, Jiaxin Yuan, Zihan Zhou, Shouli Wang, Shuo Wang, Cunliang Kong, Qi Shi, Yuxuan Li, Liner Yang, Zhiyuan Liu, Maosong Sun. Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing. 2025. Brian S. Lin, Shouli Wang, Shuo Wang 0013, Cunliang Kong, Qi Shi 0002, Liner Yang, Zhiyuan Liu 0001, Maosong Sun 0001 |
EMNLP | 11 |
| 2025 | From Unaligned to Aligned: Scaling Multilingual LLMs with Multi-Way Parallel CorporaabstractContinued pretraining and instruction tuning on large-scale multilingual data have proven to be effective in scaling large language models (LLMs) to low-resource languages.However, the unaligned nature of such data limits its ability to effectively capture cross-lingual semantics.In contrast, multi-way parallel data, where identical content is aligned across multiple languages, provides stronger cross-lingual consistency and offers greater potential for improving multilingual performance.In this paper, we introduce a large-scale, high-quality multiway parallel corpus, TED2025, based on TED Talks.The corpus spans 113 languages, with up to 50 languages aligned in parallel, ensuring extensive multilingual coverage.Using this dataset, we investigate best practices for leveraging multi-way parallel data to enhance LLMs, including strategies for continued pretraining, instruction tuning, and the analysis of key influencing factors.Experiments on six multilingual benchmarks show that models trained on multiway parallel data consistently outperform those trained on unaligned multilingual data. Yingli Shen, Wen Lai, Shuo Wang 0013, Kangyang Luo, Alexander Fraser 0001, Maosong Sun 0001 |
EMNLP | 7 |
| 2025 | GATEAU: Selecting Influential Samples for Long Context AlignmentabstractShuzheng Si, Haozhe Zhao, Gang Chen, Yunshui Li, Kangyang Luo, Chuancheng Lv, Kaikai An, Fanchao Qi, Baobao Chang, Maosong Sun. Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing. 2025. Shuzheng Si, Haozhe Zhao, Gang Chen 0039, Yunshui Li, Kangyang Luo, Chuancheng Lv, Kaikai An, Fanchao Qi, Baobao Chang, Maosong Sun 0001 |
EMNLP | 10 |
| 2025 | Looking Beyond Text: Reducing Language Bias in Large Vision-Language Models via Multimodal Dual-Attention and Soft-Image GuidanceabstractLarge vision-language models (LVLMs) have achieved impressive results in vision-language tasks.However, LVLMs suffer from hallucinations caused by language bias, which neglects images while over-relying on text.We identify two reasons for the bias: 1).Different training scales between the LLM pretraining and LVLM alignment stage.2).The learned inference bias due to short-term dependency of text data.Therefore, we propose LACING, designed to address such bias with MuLtimodal DuAlattention MeChanIsm (MDA) aNd Soft-Image Guidance (SIG).Specifically, MDA adopts a parallel dual-attention mechanism that constructs separate attention for visual and text inputs to enhance integration of visual inputs across model.SIG uses a learnable soft visual prompt during training and inference to replace visual inputs, designed to compel LVLMs to prioritize text inputs during inference.Experiments across different model architectures and scales demonstrate that LACING effectively debiases LVLMs from their language bias, enhancing visual comprehension and reducing hallucinations without additional resources. Haozhe Zhao, Shuzheng Si, Liang Chen 0024, Yichi Zhang 0010, Maosong Sun 0001, Baobao Chang, Minjia Zhang |
EMNLP | 5 |
| 2025 | Improving Generated and Retrieved Knowledge Combination Through Zero-shot GenerationabstractOpen-domain Question Answering (QA) has garnered substantial interest by combining the advantages of faithfully retrieved passages and relevant passages generated through Large Language Models (LLMs). However, there is a lack of definitive labels available to pair these sources of knowledge. In order to address this issue, we propose an unsupervised and simple framework called Bi-Reranking for Merging Generated and Retrieved Knowledge (BRMGR), which utilizes re-ranking methods for both retrieved passages and LLM-generated passages. We pair the two types of passages using two separate re-ranking methods and then combine them through greedy matching. We demonstrate that BRMGR is equivalent to employing a bipartite matching loss when assigning each retrieved passage with a corresponding LLM-generated passage. The application of our model yielded experimental results from three datasets, improving their performance by +1.7 and +1.6 on NQ and WebQ datasets, respectively, and obtaining comparable result on TriviaQA dataset when compared to competitive baselines. Xinkai Du, Quanjie Han, Yan Liu 0004, Yalin Sun, Hongbo Shan, Maosong Sun 0001 |
ICASSP | 8 |
| 2025 | Internet of Agents: Weaving a Web of Heterogeneous Agents for Collaborative IntelligenceabstractThe rapid advancement of large language models (LLMs) has paved the way for the development of highly capable autonomous agents. However, existing multi-agent frameworks often struggle with integrating diverse capable third-party agents due to reliance on agents defined within their own ecosystems. They also face challenges in simulating distributed environments, as most frameworks are limited to single-device setups. Furthermore, these frameworks often rely on hard-coded communication pipelines, limiting their adaptability to dynamic task requirements. Inspired by the concept of the Internet, we propose the Internet of Agents (IoA), a novel framework that addresses these limitations by providing a flexible and scalable platform for LLM-based multi-agent collaboration. IoA introduces an agent integration protocol, an instant-messaging-like architecture design, and dynamic mechanisms for agent teaming and conversation flow control. Through extensive experiments on general assistant tasks, embodied AI tasks, and retrieval-augmented generation benchmarks, we demonstrate that IoA consistently outperforms state-of-the-art baselines, showcasing its ability to facilitate effective collaboration among heterogeneous agents. IoA represents a step towards linking diverse agents in an Internet-like environment, where agents can seamlessly collaborate to achieve greater intelligence and capabilities. We will release our code to facilitate further research. Weize Chen, Ziming You, Yitong Guan, Cheng Yang 0002, Ruobing Xie, Zhiyuan Liu 0001, Maosong Sun 0001 |
ICLR | 10 |
| 2025 | WorkflowLLM: Enhancing Workflow Orchestration Capability of Large Language ModelsabstractRecent advancements in large language models (LLMs) have driven a revolutionary paradigm shift in process automation from Robotic Process Automation to Agentic Process Automation by automating the workflow orchestration procedure based on LLMs. However, existing LLMs (even the advanced OpenAI GPT-4o) are confined to achieving satisfactory capability in workflow orchestration. To address this limitation, we present WorkflowLLM, a data-centric framework elaborately designed to enhance the capability of LLMs in workflow orchestration. It first constructs a large-scale fine-tuning dataset WorkflowBench with 106, 763 samples, covering 1, 503 APIs from 83 applications across 28 categories. Specifically, the construction process can be divided into three phases: (1) Data Collection: we collect real-world workflow data from Apple Shortcuts and RoutineHub, transcribing them into Python-style code. We further equip them with generated hierarchical thought via GPT-4o-mini. (2) Query Expansion: we prompt GPT-4o-mini to generate more task queries to enrich the diversity and complexity of workflows. (3) Workflow Generation: we leverage an annotator model trained on collected data to generate workflows for synthesized queries. Finally, we merge the synthetic samples that pass quality confirmation with the collected samples to obtain the WorkflowBench. Based on WorkflowBench, we fine-tune Llama-3.1-8B to obtain WorkflowLlama. Our experiments show that WorkflowLlama demonstrates a strong capacity to orchestrate complex workflows, while also achieving notable generalization performance on previously unseen APIs. Additionally, WorkflowBench exhibits robust zero-shot generalization capabilities on an out-of-distribution task planning dataset, T-Eval. Our data and code are available at https://github.com/OpenBMB/WorkflowLLM. Shengda Fan, Xin Cong, Yuepeng Fu, Zhong Zhang 0004, Yuanwei Liu, Yesai Wu, Yankai Lin 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
ICLR | 10 |
| 2025 | RAG-DDR: Optimizing Retrieval-Augmented Generation Using Differentiable Data RewardsabstractRetrieval-Augmented Generation (RAG) has proven its effectiveness in mitigating hallucinations in Large Language Models (LLMs) by retrieving knowledge from external resources. To adapt LLMs for the RAG systems, current approaches use instruction tuning to optimize LLMs, improving their ability to utilize retrieved knowledge. This supervised fine-tuning (SFT) approach focuses on equipping LLMs to handle diverse RAG tasks using different instructions. However, it trains RAG modules to overfit training signals and overlooks the varying data preferences among agents within the RAG system. In this paper, we propose a Differentiable Data Rewards (DDR) method, which end-to-end trains RAG systems by aligning data preferences between different RAG modules. DDR works by collecting the rewards to optimize each agent in the RAG system with the rollout method, which prompts agents to sample some potential responses as perturbations, evaluates the impact of these perturbations on the whole RAG system, and subsequently optimizes the agent to produce outputs that improve the performance of the RAG system. Our experiments on various knowledge-intensive tasks demonstrate that DDR significantly outperforms the SFT method, particularly for LLMs with smaller-scale parameters that depend more on the retrieved knowledge. Additionally, DDR exhibits a stronger capability to align the data preference between RAG modules. The DDR method makes the generation module more effective in extracting key information from documents and mitigating conflicts between parametric memory and external knowledge. All codes are available at https://github.com/OpenMatch/RAG-DDR. Sen Mei, Zhenghao Liu 0001, Yukun Yan, Shuo Wang 0013, Shi Yu 0001, Zheni Zeng, Ge Yu 0001, Zhiyuan Liu 0001, Maosong Sun 0001, Chenyan Xiong |
ICLR | 11 |
| 2025 | Proactive Agent: Shifting LLM Agents from Reactive Responses to Active AssistanceabstractAgents powered by large language models have shown remarkable abilities in solving complex tasks. However, most agent systems remain reactive, limiting their effectiveness in scenarios requiring foresight and autonomous decision-making. In this paper, we tackle the challenge of developing proactive agents capable of anticipating and initiating tasks without explicit human instructions. We propose a novel data-driven approach for this problem. Firstly, we collect real-world human activities to generate proactive task predictions. These predictions are then labeled by human annotators as either accepted or rejected. The labeled data is used to train a reward model that simulates human judgment and serves as an automatic evaluator of the proactiveness of LLM agents. Building on this, we develop a comprehensive data generation pipeline to create a diverse dataset, ProactiveBench, containing 6,790 events. Finally, we demonstrate that fine-tuning models with the proposed ProactiveBench can significantly elicit the proactiveness of LLM agents. Experimental results show that our fine-tuned model achieves an F1-Score of 66.47% in proactively offering assistance, outperforming all open-source and close-source models. These results highlight the potential of our method in creating more proactive and effective agent systems, paving the way for future advancements in human-agent collaboration. Yaxi Lu, Shenzhi Yang, Cheng Qian 0008, Guirong Chen, Qinyu Luo, Yesai Wu, Xin Cong, Zhong Zhang 0004, Yankai Lin 0001, Weiwen Liu, Yasheng Wang, Zhiyuan Liu 0001, Fangming Liu, Maosong Sun 0001 |
ICLR | 15 |
| 2025 | A Multi-Power Law for Loss Curve Prediction Across Learning Rate SchedulesabstractTraining large models is both resource-intensive and time-consuming, making it crucial to understand the quantitative relationship between model performance and hyperparameters. In this paper, we derive an empirical law that predicts pretraining loss for large language models for every intermediate training step across various learning rate schedules, including constant, cosine, and step decay schedules. Our proposed law takes a multi-power form, combining a power law based on the sum of learning rates and additional power laws to account for a loss reduction effect as learning rate decays. We validate this law extensively on Llama-2 models of varying sizes and demonstrate that, after fitting on a few learning rate schedules, it accurately predicts the loss curves for unseen schedules of different shapes and horizons. Moreover, by minimizing the predicted final pretraining loss across learning rate schedules, we are able to find a schedule that outperforms the widely-used cosine learning rate schedule. Interestingly, this automatically discovered schedule bears some resemblance to the recently proposed Warmup-Stable-Decay (WSD) schedule (Hu et al, 2024) but achieves a slightly lower final loss. We believe these results could offer valuable insights for understanding the dynamics of pretraining and for designing learning rate schedules to improve efficiency. Kairong Luo, Haodong Wen, Shengding Hu, Zhenbo Sun, Zhiyuan Liu 0001, Maosong Sun 0001, Kaifeng Lyu |
ICLR | 6 |
| 2025 | Scaling Large Language Model-based Multi-Agent CollaborationabstractRecent breakthroughs in large language model-driven autonomous agents have revealed that multi-agent collaboration often surpasses each individual through collective reasoning. Inspired by the neural scaling law—increasing neurons enhances performance, this study explores whether the continuous addition of collaborative agents can yield similar benefits. Technically, we utilize directed acyclic graphs to organize agents into a multi-agent collaboration network (MacNet), upon which their interactive reasoning is topologically orchestrated for autonomous task solving. Extensive evaluations reveal that it effectively supports collaboration among over a thousand agents, with irregular topologies outperforming regular ones. We also identify a collaborative scaling law—the overall performance follows a logistic growth pattern as agents scale, with collaborative emergence occurring earlier than traditional neural emergence. We speculate this may be because scaling agents catalyzes their multidimensional considerations during interactive reflection and refinement, thereby producing more comprehensive artifacts. The code is available at https://github.com/OpenBMB/ChatDev/tree/macnet. Zihao Xie, Wei Liu 0161, Kunlun Zhu, Hanchen Xia, Yufan Dang, Zhuoyun Du, Weize Chen, Cheng Yang 0002, Zhiyuan Liu 0001, Maosong Sun 0001 |
ICLR | 12 |
| 2025 | Rational Decision-Making Agent with Learning Internal Utility JudgmentabstractWith remarkable advancements, large language models (LLMs) have attracted significant efforts to develop LLM-based agents capable of executing intricate multi-step decision-making tasks. Existing approaches predominantly build upon the external performance measure to guide the decision-making process but the reliance on the external performance measure as prior is problematic in real-world scenarios, where such prior may be unavailable, flawed, or even erroneous. For genuine autonomous decision-making for LLM-based agents, it is imperative to develop rationality from their posterior experiences to judge the utility of each decision independently. In this work, we propose RaDAgent (Rational Decision-Making Agent), which fosters the development of its rationality through an iterative framework involving Experience Exploration and Utility Learning. Within this framework, Elo-based Utility Learning is devised to assign Elo scores to individual decision steps to judge their utilities via pairwise comparisons. Consequently, these Elo scores guide the decision-making process to derive optimal outcomes. Experimental results on the Game of 24, WebShop, ToolBench and RestBench datasets demonstrate RaDAgent’s superiority over baselines, achieving about 7.8% improvement on average. Besides, RaDAgent also can reduce costs (ChatGPT API calls), highlighting its effectiveness and efficiency. Yining Ye, Xin Cong, Shizuo Tian, Yujia Qin, Yankai Lin 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
ICLR | 8 |
| 2025 | VisRAG: Vision-based Retrieval-augmented Generation on Multi-modality DocumentsabstractRetrieval-augmented generation (RAG) is an effective technique that enables large language models (LLMs) to utilize external knowledge sources for generation. However, current RAG systems are solely based on text, rendering it impossible to utilize vision information like layout and images that play crucial roles in real-world multi-modality documents. In this paper, we introduce VisRAG, which tackles this issue by establishing a vision-language model (VLM)-based RAG pipeline. In this pipeline, instead of first parsing the document to obtain text, the document is directly embedded using a VLM as an image and then retrieved to enhance the generation of a VLM. Compared to traditional text-based RAG, VisRAG maximizes the retention and utilization of the data information in the original documents, eliminating the information loss introduced during the parsing process. We collect both open-source and synthetic data to train the retriever in VisRAG and explore a variety of generation methods. Experiments demonstrate that VisRAG outperforms traditional RAG in both the retrieval and generation stages, achieving a 20–40% end-to-end performance gain over traditional text-based RAG pipeline. Further analysis reveals that VisRAG is efficient in utilizing training data and demonstrates strong generalization capability, positioning it as a promising solution for RAG on multi-modality documents. Our code and data are available at https://github.com/openbmb/visrag. Shi Yu 0001, Chaoyue Tang, Bokai Xu, Junbo Cui, Junhao Ran, Yukun Yan, Zhenghao Liu 0001, Shuo Wang 0013, Xu Han 0007, Zhiyuan Liu 0001, Maosong Sun 0001 |
ICLR | 11 |
| 2025 | Advancing LLM Reasoning Generalists with Preference TreesabstractWe introduce EURUS, a suite of large language models (LLMs) optimized for reasoning. Finetuned from Mistral-7B, Llama-3-8B, and Mixtral-8x22B, EURUS models achieve state-of-the-art results among open-source models on a diverse set of benchmarks covering mathematics, code generation, and logical reasoning problems. Notably, EURUX-8X22B outperforms GPT-3.5 Turbo in reasoning through a comprehensive benchmarking across 12 test sets covering five tasks. The strong performance of EURUS can be primarily attributed to ULTRAINTERACT, our newly-curated large-scale, high-quality training data dataset specifically designed for complex reasoning tasks. ULTRAINTERACT can be used in both supervised fine-tuning, preference learning, and reward modeling. It pairs each instruction with a preference tree consisting of (1) reasoning chains with diverse planning strategies in a unified format, (2) multi-turn interaction trajectories with the environment and the critique, and (3) pairwise positive and negative responses to facilitate preference learning. ULTRAINTERACT allows us to conduct an in-depth exploration of preference learning for reasoning tasks. Our investigation reveals that some well-established preference learning algorithms may be less suitable for reasoning tasks compared to their effectiveness in general conversations. The hypothesis is that in reasoning tasks, the space of correct answers is much smaller than that of incorrect ones, so it is necessary to explicitly increase the reward of chosen data. Therefore, in addition to increasing the reward margin as many preference learning algorithms do, the absolute values of positive responses’ rewards should be positive and may serve as a proxy for performance. Inspired by this, we derive a novel reward modeling objective and empirically that it leads to a stable reward modeling curve and better performance. Together with ULTRAINTERACT, we obtain a strong reward model. Lifan Yuan, Ganqu Cui, Hanbin Wang, Ning Ding 0002, Xingyao Wang 0002, Boji Shan, Zeyuan Liu, Ruobing Xie, Yankai Lin 0001, Zhenghao Liu 0001, Bowen Zhou 0002, Hao Peng 0015, Zhiyuan Liu 0001, Maosong Sun 0001 |
ICLR | 16 |
| 2025 | Sparsing Law: Towards Large Language Models with Greater Activation SparsityabstractActivation sparsity denotes the existence of substantial weakly-contributed neurons within feed-forward networks of large language models (LLMs), providing wide potential benefits such as computation acceleration. However, existing works lack thorough quantitative studies on this useful property, in terms of both its measurement and influential factors. In this paper, we address three underexplored research questions: (1) How can activation sparsity be measured more accurately? (2) How is activation sparsity affected by the model architecture and training process? (3) How can we build a more sparsely activated and efficient LLM? Specifically, we develop a generalizable and performance-friendly metric, named CETT-PPL-1%, to measure activation sparsity. Based on CETT-PPL-1%, we quantitatively study the influence of various factors and observe several important phenomena, such as the convergent power-law relationship between sparsity and training data amount, the higher competence of ReLU activation than mainstream SiLU activation, the potential sparsity merit of a small width-depth ratio, and the scale insensitivity of activation sparsity. Finally, we provide implications for building sparse and effective LLMs, and demonstrate the reliability of our findings by training a 2.4B model with a sparsity ratio of 93.52%, showing 4.1$\times$ speedup compared with its dense version. The codes and checkpoints are available at https://github.com/thunlp/SparsingLaw/. Yuqi Luo, Xu Han 0007, Yingfa Chen, Chaojun Xiao, Xiaojun Meng, Liqun Deng, Jiansheng Wei, Zhiyuan Liu 0001, Maosong Sun 0001 |
ICML | 10 |
| 2025 | NotaGen: Advancing Musicality in Symbolic Music Generation with Large Language Model Training ParadigmsabstractWe introduce NotaGen, a symbolic music generation model aiming to explore the potential of producing high-quality classical sheet music. Inspired by the success of Large Language Models (LLMs), NotaGen adopts pre-training, fine-tuning, and reinforcement learning paradigms (henceforth referred to as the LLM training paradigms). It is pre-trained on 1.6M pieces of music in ABC notation, and then fine-tuned on approximately 9K high-quality classical compositions conditioned on "period-composer-instrumentation" prompts. For reinforcement learning, we propose the CLaMP-DPO method, which further enhances generation quality and controllability without requiring human annotations or predefined rewards. Our experiments demonstrate the efficacy of CLaMP-DPO in symbolic music generation models with different architectures and encoding schemes. Furthermore, subjective A/B tests show that NotaGen outperforms baseline models against human compositions, greatly advancing musical aesthetics in symbolic music generation. Yashan Wang, Shangda Wu, Jianhuai Hu, Xingjian Du, Yueqi Peng, Yongxin Huang, Shuai Fan 0013, Feng Yu 0027, Maosong Sun 0001 |
IJCAI | 10 |
| 2025 | Benchmarking Retrieval-Augmented Generation in Multi-Modal ContextsabstractWith the rapid advancement of Multi-modal Large Language Models (MLLMs), their capability in understanding both images and text has greatly improved. However, their potential for leveraging multi-modal contextual information in Retrieval-Augmented Generation (RAG) remains largely underexplored. To address this gap, this paper introduces Multi-Modal Retrieval-Augmented Generation (M2RAG), a benchmark designed to evaluate the effectiveness of Multi-modal Large Language Models in leveraging knowledge from multi-modal retrieval documents. The benchmark comprises four tasks: image captioning, multi-modal question answering, multi-modal fact verification, and image reranking. All tasks are set in an open-domain setting, requiring RAG models to retrieve query-relevant information from a multi-modal document collection and use it as contextual input for RAG modeling. To enhance the context utilization capabilities of MLLMs, we also introduce Multi-Modal Retrieval-Augmented Instruction Tuning (MM-RAIT), an instruction tuning method that optimizes MLLMs within multi-modal contexts. Our experiments demonstrate the effectiveness of MM-RAIT by significantly improving the quality of responses generated by different RAG models, outperforming MiniCPM-V 2.6 and Qwen2-VL with 34% and 33% gains, respectively. All data and code are available at https://github.com/NEUIR/M2RAG. Zhenghao Liu 0001, Xingsheng Zhu, Tianshuo Zhou, Xiaoyuan Yi, Yukun Yan, Ge Yu 0001, Maosong Sun 0001 |
ACM Multimedia | 8 |
| 2025 | Seq1F1B: Efficient Sequence-Level Pipeline Parallelism for Large Language Model TrainingabstractSun Ao, Weilin Zhao, Xu Han, Cheng Yang, Xinrong Zhang, Zhiyuan Liu, Chuan Shi, Maosong Sun. Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers). 2025. Sun Ao, Weilin Zhao, Xu Han 0007, Cheng Yang 0002, Zhiyuan Liu 0001, Chuan Shi 0001, Maosong Sun 0001 |
NAACL (Long Papers) | 8 |
| 2025 | A Top-down Graph-based Tool for Modeling Classical Semantic Maps: A Case Study of Supplementary AdverbsabstractZhu Liu, Cunliang Kong, Ying Liu, Maosong Sun. Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers). 2025. Zhu Liu 0005, Cunliang Kong, Maosong Sun 0001 |
NAACL (Long Papers) | 4 |
| 2025 | The Overthinker's DIET: Cutting Token Calories with DIfficulty-AwarE TrainingabstractRecent large language models (LLMs) exhibit impressive reasoning but often \textit{overthink}, generating excessively long responses that hinder efficiency. We introduce DIET (DIfficulty-AwarE Training), a framework that systematically cuts these "token calories" by integrating on-the-fly problem difficulty into the reinforcement learning (RL) process. DIET dynamically adapts token compression strategies by modulating token penalty strength and conditioning target lengths on estimated task difficulty, to optimize the performance-efficiency trade-off. We also theoretically analyze the pitfalls of naive reward weighting in group-normalized RL algorithms like GRPO, and propose \textit{Advantage Weighting} technique, which enables stable and effective implementation of these difficulty-aware objectives. Experimental results demonstrate that DIET significantly reduces token counts while simultaneously improving reasoning performance. Beyond raw token reduction, we show two crucial benefits largely overlooked by prior work: (1) DIET leads to superior \textbf{inference scaling}. By maintaining high per-sample quality with fewer tokens, it enables better scaling performance via majority voting under fixed computational budgets, an area where other methods falter. (2) DIET enhances the natural positive correlation between response length and problem difficulty, ensuring verbosity is appropriately allocated, unlike many existing compression methods that disrupt this relationship. Our analyses provide a principled and effective framework for developing more efficient, practical, and high-performing LLMs. Weize Chen, Jiarui Yuan, Tailin Jin, Ning Ding 0002, Zhiyuan Liu 0001, Maosong Sun 0001 |
NeurIPS | 7 |
| 2025 | Multi-Agent Collaboration via Evolving OrchestrationabstractLarge language models (LLMs) have achieved remarkable results across diverse downstream tasks, but their monolithic nature restricts scalability and efficiency in complex problem-solving. While recent research explores multi-agent collaboration among LLMs, most approaches rely on static organizational structures that struggle to adapt as task complexity and agent numbers grow, resulting in coordination overhead and inefficiencies. To this end, we propose a puppeteer-style paradigm for LLM-based multi-agent collaboration, where a centralized orchestrator ("puppeteer") dynamically directs agents ("puppets") in response to evolving task states. This orchestrator is trained via reinforcement learning to adaptively sequence and prioritize agents, enabling flexible and evolvable collective reasoning. Experiments on closed- and open-domain scenarios show that this method achieves superior performance with reduced computational costs. Analyses further reveal that the key improvements consistently stem from the emergence of more compact, cyclic reasoning structures under the orchestrator’s evolution. Our code is available at https://github.com/OpenBMB/ChatDev/tree/puppeteer. Yufan Dang, Xueheng Luo, Jingru Fan, Zihao Xie, Ruijie Shi, Weize Chen, Cheng Yang 0002, Xiaoyin Che, Xuantang Xiong, Zhiyuan Liu 0001, Maosong Sun 0001 |
NeurIPS | 14 |
| 2025 | ParamMute: Suppressing Knowledge-Critical FFNs for Faithful Retrieval-Augmented GenerationabstractLarge language models (LLMs) integrated with retrieval-augmented generation (RAG) have improved factuality by grounding outputs in external evidence. However, they remain susceptible to unfaithful generation, where outputs contradict retrieved context despite its relevance and accuracy. Existing approaches aiming to improve faithfulness primarily focus on enhancing the utilization of external context, but often overlook the persistent influence of internal parametric knowledge during generation. In this work, we investigate the internal mechanisms behind unfaithful generation and identify a subset of mid-to-deep feed-forward networks (FFNs) that are disproportionately activated in such cases. Building on this insight, we propose Parametric Knowledge Muting through FFN Suppression (ParamMute), a framework that improves contextual faithfulness by suppressing the activation of unfaithfulness-associated FFNs and calibrating the model toward retrieved knowledge. To evaluate our approach, we introduce CoFaithfulQA, a benchmark specifically designed to evaluate faithfulness in scenarios where internal knowledge conflicts with accurate external evidence. Experimental results show that ParamMute significantly enhances faithfulness across both CoFaithfulQA and the established ConFiQA benchmark, achieving substantial reductions in reliance on parametric memory. These findings underscore the importance of mitigating internal knowledge dominance and provide a new direction for improving LLM trustworthiness in RAG. All codes are available at https://github.com/OpenBMB/ParamMute. Pengcheng Huang 0004, Zhenghao Liu 0001, Yukun Yan, Xiaoyuan Yi, Zhiyuan Liu 0001, Maosong Sun 0001, Tong Xiao 0001, Ge Yu 0001, Chenyan Xiong |
NeurIPS | 8 |
| 2025 | DCAD-2000: A Multilingual Dataset across 2000+ Languages with Data Cleaning as Anomaly DetectionabstractThe rapid development of multilingual large language models (LLMs) highlights the need for high-quality, diverse, and well-curated multilingual datasets. In this paper, we introduce DCAD-2000 (Data Cleaning as Anomaly Detection), a large-scale multilingual corpus constructed from newly extracted Common Crawl data and existing multilingual sources. DCAD-2000 covers 2,282 languages, 46.72TB of text, and 8.63 billion documents, spanning 155 high- and medium-resource languages and 159 writing scripts. To overcome the limitations of existing data cleaning approaches, which rely on manually designed heuristic thresholds, we reframe data cleaning as an anomaly detection problem. This dynamic filtering paradigm substantially improves data quality by automatically identifying and removing noisy or anomalous content. By fine-tuning LLMs on DCAD-2000, we demonstrate notable improvements in data quality, robustness of the cleaning pipeline, and downstream performance, particularly for low-resource languages across multiple multilingual benchmarks. Yingli Shen, Wen Lai, Shuo Wang 0013, Xueren Zhang, Kangyang Luo, Alexander Fraser 0001, Maosong Sun 0001 |
NeurIPS | 7 |
| 2025 | A*-Thought: Efficient Reasoning via Bidirectional Compression for Low-Resource SettingsabstractLarge Reasoning Models (LRMs) achieve superior performance by extending the thought length. However, a lengthy thinking trajectory leads to reduced efficiency. Most of the existing methods are stuck in the assumption of overthinking and attempt to reason efficiently by compressing the Chain-of-Thought, but this often leads to performance degradation. To address this problem, we introduce A*-Thought, an efficient tree search-based unified framework designed to identify and isolate the most essential thoughts from the extensive reasoning chains produced by these models. It formulates the reasoning process of LRMs as a search tree, where each node represents a reasoning span in the giant reasoning space. By combining the A* search algorithm with a cost function specific to the reasoning path, it can efficiently compress the chain of thought and determine a reasoning path with high information density and low cost. In addition, we also propose a bidirectional importance estimation mechanism, which further refines this search process and enhances its efficiency beyond uniform sampling. Extensive experiments on several advanced math tasks show that A*-Thought effectively balances performance and efficiency over a huge search space. Specifically, A*-Thought can improve the performance of QwQ-32B by 2.39$\times$ with low-budget and reduce the length of the output token by nearly 50\% with high-budget. The proposed method is also compatible with several other LRMs, demonstrating its generalization capability. The code can be accessed at: https://github.com/AI9Stars/AStar-Thought. Xiaoang Xu, Shuo Wang 0013, Zhenghao Liu 0001, Huijia Wu, Peipei Li 0002, Zhiyuan Liu 0001, Maosong Sun 0001, Zhaofeng He 0001 |
NeurIPS | 8 |
| 2025 | BurstEngine: An efficient distributed framework for training transformers On extremely Long sequences of over 1M tokensabstractExisting methods for training LLMs on long-sequence data, such as Tensor Parallelism and Context Parallelism, exhibit low Model FLOPs Utilization as sequence lengths and number of GPUs increase, especially when sequence lengths exceed 1M tokens. To address these challenges, we propose BurstEngine, an efficient framework designed to train LLMs on long-sequence data. BurstEngine introduces BurstAttention, an optimized distributed attention with lower communication cost than RingAttention. BurstAttention leverages topology-aware ring communication to fully utilize network bandwidth and incorporates fine-grained communication-computation overlap. Furthermore, BurstEngine introduces sequence-level selective checkpointing and fuses the language modeling head with the loss function to reduce memory cost. Additionally, BurstEngine introduces workload balance optimization for various types of attention masking. By integrating these optimizations, BurstEngine achieves a 1.2 × speedup with much lower memory overhead than the state-of-the-art baselines when training LLMs on extremely long sequences of over 1M tokens. Weilin Zhao, Xu Han 0007, Cheng Yang 0002, Zhiyuan Liu 0001, Chuan Shi 0001, Maosong Sun 0001 |
SC | 7 |
| 2024 | Model Composition for Multimodal Large Language ModelsabstractChi Chen, Yiyang Du, Zheng Fang, Ziyue Wang, Fuwen Luo, Peng Li, Ming Yan, Ji Zhang, Fei Huang, Maosong Sun, Yang Liu. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024. Chi Chen 0005, Yiyang Du, Ziyue Wang 0002, Fuwen Luo, Peng Li 0030, Ming Yan 0008, Ji Zhang 0011, Fei Huang 0002, Maosong Sun 0001, Yang Liu 0005 |
ACL (1) | 10 |
| 2024 | OlympiadBench: A Challenging Benchmark for Promoting AGI with Olympiad-Level Bilingual Multimodal Scientific ProblemsabstractChaoqun He, Renjie Luo, Yuzhuo Bai, Shengding Hu, Zhen Thai, Junhao Shen, Jinyi Hu, Xu Han, Yujie Huang, Yuxiang Zhang, Jie Liu, Lei Qi, Zhiyuan Liu, Maosong Sun. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024. Chaoqun He, Renjie Luo, Yuzhuo Bai, Shengding Hu, Zhen Leng Thai, Jinyi Hu, Xu Han 0007, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 14 |
| 2024 | FastFiD: Improve Inference Efficiency of Open Domain Question Answering via Sentence SelectionabstractOpen Domain Question Answering (ODQA) has been advancing rapidly in recent times, driven by significant developments in dense passage retrieval and pretrained language models.Current models typically incorporate the FiD framework, which is composed by a neural retriever alongside an encoder-decoder neural reader.In the answer generation process, the retriever will retrieve numerous passages (around 100 for instance), each of which is then individually encoded by the encoder.Subsequently, the decoder makes predictions based on these encoded passages.Nevertheless, this framework can be relatively time-consuming, particularly due to the extensive length of the gathered passages.To address this, we introduce FastFiD in this paper, a novel approach that executes sentence selection on the encoded passages.This aids in retaining valuable sentences while reducing the context length required for generating answers.Experiments on three commonly used datasets (Natural Questions, TriviaQA and ASQA) demonstrate that our method can enhance the inference speed by 2.3X-5.7X,while simultaneously maintaining the model's performance.Moreover, an in-depth analysis of the model's attention reveals that the selected sentences indeed hold a substantial contribution towards the final answer.The codes are publicly available at https://github.com Yufei Huang 0008, Xu Han 0007, Maosong Sun 0001 |
ACL (1) | 3 |
| 2024 | CODIS: Benchmarking Context-dependent Visual Comprehension for Multimodal Large Language ModelsabstractFuwen Luo, Chi Chen, Zihao Wan, Zhaolu Kang, Qidong Yan, Yingjie Li, Xiaolong Wang, Siyu Wang, Ziyue Wang, Xiaoyue Mi, Peng Li, Ning Ma, Maosong Sun, Yang Liu. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024. Fuwen Luo, Chi Chen 0005, Zihao Wan, Zhaolu Kang, Qidong Yan, Yingjie Li 0009, Xiaolong Wang 0014, Ziyue Wang 0002, Xiaoyue Mi, Peng Li 0030, Maosong Sun 0001, Yang Liu 0005 |
ACL (1) | 13 |
| 2024 | Experiential Co-Learning of Software-Developing AgentsabstractChen Qian, Yufan Dang, Jiahao Li, Wei Liu, Zihao Xie, YiFei Wang, Weize Chen, Cheng Yang, Xin Cong, Xiaoyin Che, Zhiyuan Liu, Maosong Sun. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024. Yufan Dang, Wei Liu 0161, Zihao Xie, Weize Chen, Cheng Yang 0002, Xin Cong, Xiaoyin Che, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 12 |
| 2024 | Tell Me More! Towards Implicit User Intention Understanding of Language Model Driven AgentsabstractCheng Qian, Bingxiang He, Zhong Zhuang, Jia Deng, Yujia Qin, Xin Cong, Zhong Zhang, Jie Zhou, Yankai Lin, Zhiyuan Liu, Maosong Sun. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024. Cheng Qian 0008, Bingxiang He, Zhong Zhuang, Yujia Qin, Xin Cong, Zhong Zhang 0004, Jie Zhou 0016, Yankai Lin 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 11 |
| 2024 | ChatDev: Communicative Agents for Software DevelopmentabstractChen Qian, Wei Liu, Hongzhang Liu, Nuo Chen, Yufan Dang, Jiahao Li, Cheng Yang, Weize Chen, Yusheng Su, Xin Cong, Juyuan Xu, Dahai Li, Zhiyuan Liu, Maosong Sun. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024. Wei Liu 0161, Hongzhang Liu, Yufan Dang, Cheng Yang 0002, Weize Chen, Yusheng Su, Xin Cong, Juyuan Xu, Dahai Li, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 14 |
| 2024 | Browse and Concentrate: Comprehending Multimodal Content via Prior-LLM Context FusionabstractZiyue Wang, Chi Chen, Yiqi Zhu, Fuwen Luo, Peng Li, Ming Yan, Ji Zhang, Fei Huang, Maosong Sun, Yang Liu. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024. Ziyue Wang 0002, Chi Chen 0005, Yiqi Zhu, Fuwen Luo, Peng Li 0030, Ming Yan 0008, Ji Zhang 0011, Fei Huang 0002, Maosong Sun 0001, Yang Liu 0005 |
ACL (1) | 9 |
| 2024 | LoRA-Flow: Dynamic LoRA Fusion for Large Language Models in Generative TasksabstractLoRA employs lightweight modules to customize large language models (LLMs) for each downstream task or domain, where different learned additional modules represent diverse skills.Combining existing LoRA modules to address new tasks can enhance the reusability of learned LoRA modules, particularly beneficial for tasks with limited annotated data.Most prior works on LoRA combination primarily rely on task-level weights for each involved LoRA, making different examples and tokens share the same LoRA weights.However, in generative tasks, different tokens may necessitate diverse skills to manage.Taking the Chinese math task as an example, understanding the problem description may depend more on the Chinese LoRA, while the calculation part may rely more on the math LoRA.To this end, we propose LoRA-Flow, which utilizes dynamic weights to adjust the impact of different LoRA modules.The weights at each step are determined by a fusion gate with extremely few parameters, which can be learned with only 200 training examples.Experiments across six generative tasks demonstrate that our method consistently outperforms baselines with tasklevel fusion weights.This underscores the necessity of introducing dynamic fusion weights for LoRA combination. 1 Hanqing Wang 0003, Bowen Ping, Shuo Wang 0013, Xu Han 0007, Yun Chen 0007, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 7 |
| 2024 | UltraLink: An Open-Source Knowledge-Enhanced Multilingual Supervised Fine-tuning DatasetabstractHaoyu Wang, Shuo Wang, Yukun Yan, Xujia Wang, Zhiyu Yang, Yuzhuang Xu, Zhenghao Liu, Liner Yang, Ning Ding, Xu Han, Zhiyuan Liu, Maosong Sun. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024. Shuo Wang 0013, Yukun Yan, Xujia Wang, Zhiyu Yang 0001, Yuzhuang Xu, Zhenghao Liu 0001, Liner Yang, Ning Ding 0002, Xu Han 0007, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 12 |
| 2024 | Reasoning in Conversation: Solving Subjective Tasks through Dialogue Simulation for Large Language ModelsabstractXiaolong Wang, Yile Wang, Yuanchi Zhang, Fuwen Luo, Peng Li, Maosong Sun, Yang Liu. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024. Xiaolong Wang 0014, Yile Wang 0001, Yuanchi Zhang, Fuwen Luo, Peng Li 0030, Maosong Sun 0001, Yang Liu 0005 |
ACL (1) | 6 |
| 2024 | ınftyBench: Extending Long Context Evaluation Beyond 100K TokensabstractXinrong Zhang, Yingfa Chen, Shengding Hu, Zihang Xu, Junhao Chen, Moo Hao, Xu Han, Zhen Thai, Shuo Wang, Zhiyuan Liu, Maosong Sun. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024. Yingfa Chen, Shengding Hu, Zihang Xu, Moo Khai Hao, Xu Han 0007, Zhen Leng Thai, Shuo Wang 0013, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 11 |
| 2024 | Enhancing Multilingual Capabilities of Large Language Models through Self-Distillation from Resource-Rich LanguagesabstractYuanchi Zhang, Yile Wang, Zijun Liu, Shuo Wang, Xiaolong Wang, Peng Li, Maosong Sun, Yang Liu. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024. Yuanchi Zhang, Yile Wang 0001, Shuo Wang 0013, Xiaolong Wang 0014, Peng Li 0030, Maosong Sun 0001, Yang Liu 0005 |
ACL (1) | 7 |
| 2024 | Empowering Private Tutoring by Chaining Large Language ModelsabstractArtificial intelligence has been applied in various aspects of online education to facilitate teaching and learning. However, few approaches have been made towards a complete AI-powered tutoring system. In this work, we explore the development of a full-fledged intelligent tutoring system based on large language models (LLMs). The proposed system ChatTutor, powered by state-of-the-art LLMs, is equipped with automatic course planning and adjusting, informative instruction, and adaptive quiz offering and evaluation. ChatTutor is decomposed into three inter-connected core processes: interaction, reflection, and reaction. Each process is implemented by chaining LLM-powered tools along with dynamically updated memory modules. To demonstrate the mechanism of each working module and the benefits of structured memory control and adaptive reflection, we conduct a wide range of analysis based on statistical results and user study. The analysis shows the designed processes boost system consistency and stability under long-term interaction and intentional disruptions, with up to 5% and 20% increase in performance respectively. Meanwhile, we also compare the system with scripts from real-world online learning platform and discuss the potential issues unique to LLM-based systems. Yulin Chen 0001, Ning Ding 0002, Hai-Tao Zheng 0002, Zhiyuan Liu 0001, Maosong Sun 0001, Bowen Zhou 0002 |
CIKM | 5 |
| 2024 | Robust and Scalable Model Editing for Large Language ModelsabstractLarge language models (LLMs) can make predictions using parametric knowledge – knowledge encoded in the model weights – or contextual knowledge – knowledge presented in the context. In many scenarios, a desirable behavior is that LLMs give precedence to contextual knowledge when it conflicts with the parametric knowledge, and fall back to using their parametric knowledge when the context is irrelevant. This enables updating and correcting the model’s knowledge by in-context editing instead of retraining. Previous works have shown that LLMs are inclined to ignore contextual knowledge and fail to reliably fall back to parametric knowledge when presented with irrelevant context. In this work, we discover that, with proper prompting methods, instruction-finetuned LLMs can be highly controllable by contextual knowledge and robust to irrelevant context. Utilizing this feature, we propose EREN (Edit models by REading Notes) to improve the scalability and robustness of LLM editing. To better evaluate the robustness of model editors, we collect a new dataset, that contains irrelevant questions that are more challenging than the ones in existing datasets. Empirical results show that our method outperforms current state-of-the-art methods by a large margin. Unlike existing techniques, it can integrate knowledge from multiple edits, and correctly respond to syntactically similar but semantically unrelated inputs (and vice versa). The source code can be found at https://github.com/thunlp/EREN. Yingfa Chen, Zhengyan Zhang, Xu Han 0007, Chaojun Xiao, Zhiyuan Liu 0001, Kuai Li, Maosong Sun 0001 |
LREC/COLING | 9 |
| 2024 | Fine-Grained Legal Argument-Pair Extraction via Coarse-Grained Pre-trainingabstractLegal Argument-Pair Extraction (LAE) is dedicated to the identification of interactive arguments targeting the same subject matter within legal complaints and corresponding defenses. This process serves as a foundation for automatically recognizing the focal points of disputes. Current methodologies predominantly conceptualize LAE as a supervised sentence-pair classification problem and usually necessitate extensive manual annotations, thereby constraining their scalability and general applicability. To this end, we present an innovative approach to LAE that focuses on fine-grained alignment of argument pairs, building upon coarse-grained complaint-defense pairs. This strategy stems from two key observations: 1) In general, every argument presented in a legal complaint is likely to be addressed by at least one corresponding argument in the defense. 2) It’s rare for multiple complaint arguments to be addressed by a single defense argument; rather, each complaint argument usually corresponds to a unique defense argument. Motivated by these insights, we develop a specialized pre-training framework. Our model employs pre-training objectives designed to exploit the coarse-grained supervision signals. This enables expressive representations of legal arguments for LAE, even when working with a limited amount of labeled data. To verify the effectiveness of our model, we construct the largest LAE datasets from two representative causes, private lending, and contract dispute. The experimental results demonstrate that our model can effectively capture informative argument knowledge from unlabeled complaint-defense pairs and outperform the unsupervised and supervised baselines by 3.7 and 2.4 points on average respectively. Besides, our model can reach superior accuracy with only half manually annotated data. The datasets and code can be found in https://github.com/thunlp/LAE. Chaojun Xiao, Yutao Sun, Yuan Yao 0013, Zhiyuan Liu 0001, Maosong Sun 0001 |
LREC/COLING | 7 |
| 2024 | RLHF-V: Towards Trustworthy MLLMs via Behavior Alignment from Fine-Grained Correctional Human FeedbackabstractMultimodal Large Language Models (MLLMs) have recently demonstrated impressive capabilities in multimodal understanding, reasoning, and interaction. However, existing MLLMs prevalently suffer from serious hallucination problems, generating text that is not factually grounded in associated images. The problem makes existing MLLMs untrustworthy and thus impractical in real-world (especially high-stakes) applications. To address the challenge, we present RLHF-V, which enhances MLLM trustworthiness via behavior alignment from fine-grained correctional human feedback. Specifically, RLHF-V collects human preference in the form of segment-level corrections on hallucinations, and performs dense direct preference optimization over the human feedback. Comprehensive experiments on five benchmarks in both automatic and human evaluation show that, RLHF-V can enable substantially more trustworthy MLLM behaviors with promising data and computation efficiency. Remarkably, using 1.4k annotated data samples, RLHF-V significantly reduces the hallucination rate of the base MLLM by 34.8%, outperforming the concurrent LLaVA-RLHF trained on 10k annotated data. The final model achieves state-of-the-art performance in trustwor-thiness among open-source MLLMs, and shows better ro-bustness than GPT-4V in preventing hallucinations aroused from over-generalization. Tianyu Yu 0002, Yuan Yao 0013, Haoye Zhang, Taiwen He, Yifeng Han, Ganqu Cui, Jinyi Hu, Zhiyuan Liu 0001, Hai-Tao Zheng 0002, Maosong Sun 0001 |
CVPR | 10 |
| 2024 | Enhancing Legal Case Retrieval via Scaling High-quality Synthetic Query-Candidate PairsabstractLegal case retrieval (LCR) aims to provide similar cases as references for a given fact description.This task is crucial for promoting consistent judgments in similar cases, effectively enhancing judicial fairness and improving work efficiency for judges.However, existing works face two main challenges for real-world applications: existing works mainly focus on case-to-case retrieval using lengthy queries, which does not match real-world scenarios; and the limited data scale, with current datasets containing only hundreds of queries, is insufficient to satisfy the training requirements of existing data-hungry neural models.To address these issues, we introduce an automated method to construct synthetic query-candidate pairs and build the largest LCR dataset to date, LEAD, which is hundreds of times larger than existing datasets.This data construction method can provide ample training signals for LCR models.Experimental results demonstrate that model training with our constructed data can achieve state-of-the-art results on two widely-used LCR benchmarks.Besides, the construction method can also be applied to civil cases and achieve promising results.The data and codes can be found in https://github.com/thunlp/LEAD. Chaojun Xiao, Zhenghao Liu 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
EMNLP | 6 |
| 2024 | Controllable Preference Optimization: Toward Controllable Multi-Objective AlignmentabstractYiju Guo, Ganqu Cui, Lifan Yuan, Ning Ding, Zexu Sun, Bowen Sun, Huimin Chen, Ruobing Xie, Jie Zhou, Yankai Lin, Zhiyuan Liu, Maosong Sun. Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing. 2024. Yiju Guo, Ganqu Cui, Lifan Yuan, Ning Ding 0002, Zexu Sun, Ruobing Xie, Jie Zhou 0016, Yankai Lin 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
EMNLP | 12 |
| 2024 | Beyond the Turn-Based Game: Enabling Real-Time Conversations with Duplex ModelsabstractXinrong Zhang, Yingfa Chen, Shengding Hu, Xu Han, Zihang Xu, Yuanwei Xu, Weilin Zhao, Maosong Sun, Zhiyuan Liu. Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing. 2024. Yingfa Chen, Shengding Hu, Xu Han 0007, Zihang Xu, Yuanwei Xu, Weilin Zhao, Maosong Sun 0001, Zhiyuan Liu 0001 |
EMNLP | 8 |
| 2024 | Ouroboros: Generating Longer Drafts Phrase by Phrase for Faster Speculative DecodingabstractWeilin Zhao, Yuxiang Huang, Xu Han, Wang Xu, Chaojun Xiao, Xinrong Zhang, Yewei Fang, Kaihuo Zhang, Zhiyuan Liu, Maosong Sun. Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing. 2024. Weilin Zhao, Yuxiang Huang 0001, Xu Han 0007, Chaojun Xiao, Yewei Fang, Kaihuo Zhang, Zhiyuan Liu 0001, Maosong Sun 0001 |
EMNLP | 10 |
| 2024 | DecorateLM: Data Engineering through Corpus Rating, Tagging, and Editing with Language ModelsabstractRanchi Zhao, Zhen Leng Thai, Yifan Zhang, Shengding Hu, Jie Zhou, Yunqi Ba, Jie Cai, Zhiyuan Liu, Maosong Sun. Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing. 2024. Ranchi Zhao, Zhen Leng Thai, Shengding Hu, Jie Zhou 0024, Yunqi Ba, Jie Cai 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
EMNLP | 9 |
| 2024 | Label Dependencies-Aware Set Prediction Networks for Multi-Label Text ClassificationabstractMulti-label text classification involves extracting all relevant labels from a sentence. Given the unordered nature of these labels, we propose approaching the problem as a set prediction task. To address the correlation between labels, we leverage Graph Convolutional Networks and construct an adjacency matrix based on the statistical relations between labels. Additionally, we enhance recall ability by applying the Bhattacharyya distance to the output distributions of the set prediction networks. We evaluate the effectiveness of our approach on two multi-label datasets and demonstrate its superiority over previous baselines through experimental results. Xinkai Du, Quanjie Han, Yalin Sun, Maosong Sun 0001 |
ICASSP | 5 |
| 2024 | AgentVerse: Facilitating Multi-Agent Collaboration and Exploring Emergent BehaviorsabstractAutonomous agents empowered by Large Language Models (LLMs) have undergone significant improvements, enabling them to generalize across a broad spectrum of tasks. However, in real-world scenarios, cooperation among individuals is often required to enhance the efficiency and effectiveness of task accomplishment. Hence, inspired by human group dynamics, we propose a multi-agent framework AgentVerse that can effectively orchestrate a collaborative group of expert agents as a greater-than-the-sum-of-its-parts system. Our experiments demonstrate that AgentVerse can proficiently deploy multi-agent groups that outperform a single agent. Extensive experiments on text understanding, reasoning, coding, tool utilization, and embodied AI confirm the effectiveness of AgentVerse. Moreover, our analysis of agent interactions within AgentVerse reveals the emergence of specific collaborative behaviors, contributing to heightened group efficiency. We will release our codebase, AgentVerse, to further facilitate multi-agent research. Weize Chen, Yusheng Su, Jingwei Zuo, Cheng Yang 0002, Chenfei Yuan, Chi-Min Chan, Heyang Yu, Yaxi Lu, Yi-Hsin Hung, Yujia Qin, Xin Cong, Ruobing Xie, Zhiyuan Liu 0001, Maosong Sun 0001, Jie Zhou 0016 |
ICLR | 15 |
| 2024 | Large Multilingual Models Pivot Zero-Shot Multimodal Learning across LanguagesabstractRecently there has been a significant surge in multimodal learning in terms of both image-to-text and text-to-image generation. However, the success is typically limited to English, leaving other languages largely behind. Building a competitive counterpart in other languages is highly challenging due to the low-resource nature of non-English multimodal data (i.e., lack of large-scale, high-quality image-text data). In this work, we propose MPM, an effective training paradigm for training large multimodal models in low-resource languages. MPM demonstrates that Multilingual language models can Pivot zero-shot Multimodal learning across languages. Specifically, based on a strong multilingual large language model, multimodal models pretrained on English-only image-text data can well generalize to other languages in a (quasi)-zero-shot manner, even surpassing models trained on image-text data in native languages. Taking Chinese as a practice of MPM, we build large multimodal models VisCPM in image-to-text and text-to-image generation, which achieve state-of-the-art (open-source) performance in Chinese. To facilitate future research, we open-source codes and model weights at https://github.com/OpenBMB/VisCPM. Jinyi Hu, Yuan Yao 0013, Chongyi Wang, Shan Wang 0015, Yinxu Pan, Tianyu Yu 0002, Hanghao Wu, Haoye Zhang, Xu Han 0007, Yankai Lin 0001, Jiao Xue, Dahai Li, Zhiyuan Liu 0001, Maosong Sun 0001 |
ICLR | 16 |
| 2024 | Predicting Emergent Abilities with Infinite Resolution EvaluationabstractThe scientific scale-up of large language models (LLMs) necessitates a comprehensive understanding of their scaling properties. However, the existing literature on the scaling properties only yields an incomplete answer: optimization loss decreases predictably as the model size increases, in line with established scaling law; yet no scaling law for task has been established and the task performances are far from predictable during scaling. Task performances typically show minor gains on small models until they improve dramatically once models exceed a size threshold, exemplifying the ''emergent abilities''. In this study, we discover that small models, although they exhibit minor performance, demonstrate critical and consistent task performance improvements that are not captured by conventional evaluation strategies due to insufficient measurement resolution. To measure such improvements, we introduce PassUntil, an evaluation strategy with theoretically infinite resolution, through massive sampling in the decoding phase. With PassUntil, we conduct a quantitative investigation into the scaling law of task performance. The investigation contains two parts. Firstly, a strict task scaling law that is not conventionally known to exist, is identified, enhancing the predictability of task performances. Remarkably, we are able to predict the performance of the 2.4B model on code generation with merely 0.05\% deviation before training starts, which is the first systematic attempt to verify predictable scaling proposed by GPT-4's report. Secondly, underpinned by PassUntil, we are able to study emergent abilities quantitatively. We identify a kind of accelerated emergence whose scaling curve cannot be fitted by standard scaling law function and has a increasing speed. We then examine two hypothesis and imply that the ``multiple circuits hypothesis'' might be responsible for the accelerated emergence. Shengding Hu, Xin Liu 0086, Xu Han 0007, Chaoqun He, Weilin Zhao, Yankai Lin 0001, Ning Ding 0002, Zebin Ou, Guoyang Zeng, Zhiyuan Liu 0001, Maosong Sun 0001 |
ICLR | 12 |
| 2024 | ToolLLM: Facilitating Large Language Models to Master 16000+ Real-world APIsabstractDespite the advancements of open-source large language models (LLMs), e.g., LLaMA, they remain significantly limited in tool-use capabilities, i.e., using external tools (APIs) to fulfill human instructions. The reason is that current instruction tuning largely focuses on basic language tasks but ignores the tool-use domain. This is in contrast to the excellent tool-use capabilities of state-of-the-art (SOTA) closed-source LLMs, e.g., ChatGPT. To bridge this gap, we introduce ToolLLM, a general tool-use framework encompassing data construction, model training, and evaluation. We first present ToolBench, an instruction-tuning dataset for tool use, which is constructed automatically using ChatGPT. Specifically, the construction can be divided into three stages: (i) API collection: we collect 16,464 real-world RESTful APIs spanning 49 categories from RapidAPI Hub; (ii) instruction generation: we prompt ChatGPT to generate diverse instructions involving these APIs, covering both single-tool and multi-tool scenarios; (iii) solution path annotation: we use ChatGPT to search for a valid solution path (chain of API calls) for each instruction. To enhance the reasoning capabilities of LLMs, we develop a novel depth-first search-based decision tree algorithm. It enables LLMs to evaluate multiple reasoning traces and expand the search space. Moreover, to evaluate the tool-use capabilities of LLMs, we develop an automatic evaluator: ToolEval. Based on ToolBench, we fine-tune LLaMA to obtain an LLM ToolLLaMA, and equip it with a neural API retriever to recommend appropriate APIs for each instruction. Experiments show that ToolLLaMA demonstrates a remarkable ability to execute complex instructions and generalize to unseen APIs, and exhibits comparable performance to ChatGPT. Our ToolLLaMA also demonstrates strong zero-shot generalization ability in an out-of-distribution tool-use dataset: APIBench. Yujia Qin, Shihao Liang, Yining Ye, Kunlun Zhu, Lan Yan, Yaxi Lu, Yankai Lin 0001, Xin Cong, Xiangru Tang, Bill Qian, Sihan Zhao, Lauren Hong, Runchu Tian, Ruobing Xie, Jie Zhou 0016, Mark Gerstein, Dahai Li, Zhiyuan Liu 0001, Maosong Sun 0001 |
ICLR | 19 |
| 2024 | ULTRAFEEDBACK: Boosting Language Models with Scaled AI FeedbackabstractLearning from human feedback has become a pivot technique in aligning large language models (LLMs) with human preferences. However, acquiring vast and premium human feedback is bottlenecked by time, labor, and human capability, resulting in small sizes or limited topics of current datasets. This further hinders feedback learning as well as alignment research within the open-source community. To address this issue, we explore how to go beyond human feedback and collect high-quality AI feedback automatically for a scalable alternative. Specifically, we identify scale and diversity as the key factors for feedback data to take effect. Accordingly, we first broaden instructions and responses in both amount and breadth to encompass a wider range of user-assistant interactions. Then, we meticulously apply a series of techniques to mitigate annotation biases for more reliable AI feedback. We finally present UltraFeedback, a large-scale, high-quality, and diversified AI feedback dataset, which contains over 1 million GPT-4 feedback for 250k user-assistant conversations from various aspects. Built upon UltraFeedback, we align a LLaMA-based model by best-of-$n$ sampling and reinforcement learning, demonstrating its exceptional performance on chat benchmarks. Our work validates the effectiveness of scaled AI feedback data in constructing strong open-source chat language models, serving as a solid foundation for future feedback learning research. Ganqu Cui, Lifan Yuan, Ning Ding 0002, Guanming Yao, Bingxiang He, Wei Zhu 0016, Yuan Ni, Guo Tong Xie, Ruobing Xie, Yankai Lin 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
ICML | 12 |
| 2024 | Exploring the Benefit of Activation Sparsity in Pre-trainingabstractPre-trained Transformers inherently possess the characteristic of sparse activation, where only a small fraction of the neurons are activated for each token. While sparse activation has been explored through post-training methods, its potential in pre-training remains untapped. In this work, we first study how activation properties change during pre-training. Our examination reveals that Transformers exhibit sparse activation throughout the majority of the pre-training process while the activation correlation keeps evolving as training progresses. Leveraging this observation, we propose Switchable Sparse-Dense Learning (SSD). SSD adaptively switches between the Mixtures-of-Experts (MoE) based sparse training and the conventional dense training during the pre-training process, leveraging the efficiency of sparse training and avoiding the static activation correlation of sparse training. Compared to dense training, SSD achieves comparable performance with identical model size and reduces pre-training costs. Moreover, the models trained with SSD can be directly used as MoE models for sparse inference and achieve the same performance as dense models with up to $2\times$ faster inference speed. Codes are available at https://github.com/thunlp/moefication. Zhengyan Zhang, Chaojun Xiao, Qiujieli Qin, Yankai Lin 0001, Xu Han 0007, Zhiyuan Liu 0001, Ruobing Xie, Maosong Sun 0001, Jie Zhou 0016 |
ICML | 9 |
| 2024 | On the Essence and Prospect: An Investigation of Alignment Approaches for Big Models
Xinpeng Wang 0001, Shitong Duan, Xiaoyuan Yi, Jing Yao 0003, Shanlin Zhou, Zhihua Wei 0001, Peng Zhang 0060, Dongkuan Xu, Maosong Sun 0001, Xing Xie 0001 |
IJCAI | 9 |
| 2024 | Can Large Language Models Analyze Graphs like Professionals? A Benchmark, Datasets and ModelsabstractThe need to analyze graphs is ubiquitous across various fields, from social networks to biological research and recommendation systems. Therefore, enabling the ability of large language models (LLMs) to process graphs is an important step toward more advanced general intelligence. However, current LLM benchmarks on graph analysis require models to directly reason over the prompts describing graphtopology, and are thus limited to small graphs with only a few dozens of nodes. In contrast, human experts typically write programs based on popular libraries for task solving, and can thus handle graphs with different scales. To this end, a question naturally arises: can LLMs analyze graphs like professionals? In this paper, we introduce ProGraph, a manually crafted benchmark containing 3 categories of graph tasks. The benchmark expects solutions based on programming instead of directly reasoning over raw inputs. Our findings reveal that the performance of current LLMs is unsatisfactory, with the best model achieving only 36% accuracy. To bridge this gap, we propose LLM4Graph datasets, which include crawled documents and auto-generated codes based on 6 widely used graph libraries. By augmenting closed-source LLMs with document retrieval and fine-tuning open-source ones on the codes, we show 11-32% absolute improvements in their accuracies. Our results underscore that the capabilities of LLMs in handling structured data are still under-explored, and show the effectiveness of LLM4Graph in enhancing LLMs’ proficiency of graph analysis. The benchmark, datasets and enhanced open-sourcemodels are available at https://github.com/BUPT-GAMMA/ProGraph. Xin Li 0005, Weize Chen, Qizhi Chu, Zhaojun Sun, Chuan Shi 0001, Zhiyuan Liu 0001, Maosong Sun 0001, Cheng Yang 0002 |
NeurIPS | 11 |
| 2024 | Delta-CoMe: Training-Free Delta-Compression with Mixed-Precision for Large Language ModelsabstractFine-tuning is a crucial process for adapting large language models (LLMs) to diverse applications. In certain scenarios, such as multi-tenant serving, deploying multiple LLMs becomes necessary to meet complex demands. Recent studies suggest decomposing a fine-tuned LLM into a base model and corresponding delta weights, which are then compressed using low-rank or low-bit approaches to reduce costs. In this work, we observe that existing low-rank and low-bit compression methods can significantly harm the model performance for task-specific fine-tuned LLMs (e.g., WizardMath for math problems). Motivated by the long-tail distribution of singular values in the delta weights, we propose a delta quantization approach using mixed-precision. This method employs higher-bit representation for singular vectors corresponding to larger singular values. We evaluate our approach on various fine-tuned LLMs, including math LLMs, code LLMs, chat LLMs, and even VLMs. Experimental results demonstrate that our approach performs comparably to full fine-tuned LLMs, surpassing both low-rank and low-bit baselines by a considerable margin. Additionally, we show that our method is compatible with various backbone LLMs, such as Llama-2, Llama-3, and Mistral, highlighting its generalizability. Bowen Ping, Shuo Wang 0013, Hanqing Wang 0003, Xu Han 0007, Yuzhuang Xu, Yukun Yan, Yun Chen 0007, Baobao Chang, Zhiyuan Liu 0001, Maosong Sun 0001 |
NeurIPS | 10 |
| 2024 | InfLLM: Training-Free Long-Context Extrapolation for LLMs with an Efficient Context MemoryabstractLarge language models (LLMs) have emerged as a cornerstone in real-world applications with lengthy streaming inputs (e.g., LLM-driven agents). However, existing LLMs, pre-trained on sequences with a restricted maximum length, cannot process longer sequences due to the out-of-domain and distraction issues. Common solutions often involve continual pre-training on longer sequences, which will introduce expensive computational overhead and uncontrollable change in model capabilities. In this paper, we unveil the intrinsic capacity of LLMs for understanding extremely long sequences without any fine-tuning. To this end, we introduce a training-free memory-based method, InfLLM. Specifically, InfLLM stores distant contexts into additional memory units and employs an efficient mechanism to lookup token-relevant units for attention computation. Thereby, InfLLM allows LLMs to efficiently process long sequences with a limited context window and well capture long-distance dependencies. Without any training, InfLLM enables LLMs that are pre-trained on sequences consisting of a few thousand tokens to achieve comparable performance with competitive baselines that continually train these LLMs on long sequences. Even when the sequence length is scaled to 1,024K, InfLLM still effectively captures long-distance dependencies. Our code can be found at https://github.com/thunlp/InfLLM. Chaojun Xiao, Pengle Zhang, Xu Han 0007, Guangxuan Xiao, Yankai Lin 0001, Zhengyan Zhang, Zhiyuan Liu 0001, Maosong Sun 0001 |
NeurIPS | 8 |
| 2024 | scMulan: A Multitask Generative Pre-Trained Language Model for Single-Cell Analysis
Haiyang Bian, Xiaomin Dong, Chen Li 0001, Minsheng Hao, Jinyi Hu, Maosong Sun 0001, Lei Wei 0009, Xuegong Zhang |
RECOMB | 8 |
| 2024 | ChatMol: interactive molecular discovery with natural languageabstractMOTIVATION: Natural language is poised to become a key medium for human-machine interactions in the era of large language models. In the field of biochemistry, tasks such as property prediction and molecule mining are critically important yet technically challenging. Bridging molecular expressions in natural language and chemical language can significantly enhance the interpretability and ease of these tasks. Moreover, it can integrate chemical knowledge from various sources, leading to a deeper understanding of molecules. RESULTS: Recognizing these advantages, we introduce the concept of conversational molecular design, a novel task that utilizes natural language to describe and edit target molecules. To better accomplish this task, we develop ChatMol, a knowledgeable and versatile generative pretrained model. This model is enhanced by incorporating experimental property information, molecular spatial knowledge, and the associations between natural and chemical languages. Several typical solutions including large language models (e.g. ChatGPT) are evaluated, proving the challenge of conversational molecular design and the effectiveness of our knowledge enhancement approach. Case observations and analysis offer insights and directions for further exploration of natural-language interaction in molecular discovery. AVAILABILITY AND IMPLEMENTATION: Codes and data are provided in https://github.com/Ellenzzn/ChatMol/tree/main. Zheni Zeng, Bangchen Yin, Cheng Yang 0002, Haishen Yao, Xingzhi Sun 0002, Maosong Sun 0001, Guo Tong Xie, Zhiyuan Liu 0001 |
Bioinform. | 8 |
| 2024 | Hyperbolic Pre-Trained Language ModelabstractIn recent years, we have witnessed significant improvements in pre-trained language models (PLM) brought about by the scaling of parameter sizes and data amounts. However, this also brings high computational and storage costs. In this paper, we present a new direction to improve PLMs without scaling parameters and data: adopting a geometric feature space that is more suitable for encoding the intrinsic structured features of text. Although text is generally considered unstructured data, it possesses rich intrinsic structured features that signify syntactic and semantic relationships. Leveraging these structured features is vital for text understanding. Given that structured features are better encoded in hyperbolic spaces than in the Euclidean spaces used by conventional PLMs, we propose that PLMs should operate entirely within hyperbolic spaces. Our experiments demonstrate the superiority of hyperbolic PLMs over Euclidean PLMs across a wide variety of tasks, using the same parameter and data settings. This suggests that altering the geometry of model representation is a promising direction for model enhancement. The code is released athttps://github.com/thunlp/hyperbolic_llm Weize Chen, Xu Han 0007, Yankai Lin 0001, Kaichen He, Ruobing Xie, Jie Zhou 0016, Zhiyuan Liu 0001, Maosong Sun 0001 |
IEEE ACM Trans. Audio Speech Lang. Process. | 8 |
| 2024 | Exploring Universal Intrinsic Task Subspace for Few-Shot Learning via Prompt TuningabstractWhy can pre-trained language models (PLMs) learn universal representations and effectively adapt to broad NLP tasks differing a lot superficially? In this work, we empirically find evidence indicating that the adaptations of PLMs to various few-shot tasks can be reparameterized as optimizing only a few free parameters in a unified low-dimensionalintrinsic task subspace, which may help us understand why PLMs could easily adapt to various NLP tasks with small-scale data. To find such a subspace and examine its universality, we propose an analysis pipeline calledintrinsic prompt tuning(IPT). Specifically, we resort to the recent success of prompt tuning and decompose the soft prompts of multiple NLP tasks into the same low-dimensional nonlinear subspace, then we learn to adapt the PLM to unseen data or tasks by only tuning parameters in this subspace. In the experiments, we study diverse few-shot NLP tasks and surprisingly find that in a 250-dimensional subspace found with 100 tasks, by only tuning 250 free parameters, we can recover 97% and 83% of the full prompt tuning performance for 100 seen tasks (using different training data) and 20 unseen tasks, respectively, showing great generalization ability of the found intrinsic task subspace. Besides being an analysis tool, IPTcould further help us improve the prompt tuning stability. Yujia Qin, Xiaozhi Wang, Yusheng Su, Yankai Lin 0001, Ning Ding 0002, Jing Yi, Weize Chen, Zhiyuan Liu 0001, Juan-Zi Li, Lei Hou 0001, Peng Li 0030, Maosong Sun 0001, Jie Zhou 0016 |
IEEE ACM Trans. Audio Speech Lang. Process. | 12 |
| 2024 | Personality-affected Emotion Generation in Dialog SystemsabstractGenerating appropriate emotions for responses is essential for dialogue systems to provide human-like interaction in various application scenarios. Most previous dialogue systems tried to achieve this goal by learning empathetic manners from anonymous conversational data. However, emotional responses generated by those methods may be inconsistent, which will decrease user engagement and service quality. Psychological findings suggest that the emotional expressions of humans are rooted in personality traits. Therefore, we propose a new task, Personality-affected Emotion Generation, to generate emotion based on the personality given to the dialogue system and further investigate a solution through the personality-affected mood transition. Specifically, we first construct a daily dialogue dataset, Personality EmotionLines Dataset ( PELD ), with emotion and personality annotations. Subsequently, we analyze the challenges in this task, i.e., (1) heterogeneously integrating personality and emotional factors and (2) extracting multi-granularity emotional information in the dialogue context. Finally, we propose to model the personality as the transition weight by simulating the mood transition process in the dialogue system and solve the challenges above. We conduct extensive experiments on PELD for evaluation. Results suggest that by adopting our method, the emotion generation performance is improved by 13% in macro-F1 and 5% in weighted-F1 from the BERT-base model. Jiannong Cao 0001, Jiaxing Shen, Ruosong Yang, Shuaiqi Liu 0002, Maosong Sun 0001 |
ACM Trans. Inf. Syst. | 6 |
| 2023 | Visually Grounded Commonsense Knowledge AcquisitionabstractLarge-scale commonsense knowledge bases empower a broad range of AI applications, where the automatic extraction of commonsense knowledge (CKE) is a fundamental and challenging problem. CKE from text is known for suffering from the inherent sparsity and reporting bias of commonsense in text. Visual perception, on the other hand, contains rich commonsense knowledge about real-world entities, e.g., (person, can_hold, bottle), which can serve as promising sources for acquiring grounded commonsense knowledge. In this work, we present CLEVER, which formulates CKE as a distantly supervised multi-instance learning problem, where models learn to summarize commonsense relations from a bag of images about an entity pair without any human annotation on image instances. To address the problem, CLEVER leverages vision-language pre-training models for deep understanding of each image in the bag, and selects informative instances from the bag to summarize commonsense entity relations via a novel contrastive attention mechanism. Comprehensive experimental results in held-out and human evaluation show that CLEVER can extract commonsense knowledge in promising quality, outperforming pre-trained language model-based methods by 3.9 AUC and 6.4 mAUC points. The predicted commonsense scores show strong correlation with human judgment with a 0.78 Spearman coefficient. Moreover, the extracted commonsense can also be grounded into images with reasonable interpretability. The data and codes can be obtained at https://github.com/thunlp/CLEVER. Yuan Yao 0013, Tianyu Yu 0002, Mengdi Li 0006, Ruobing Xie, Cornelius Weber, Zhiyuan Liu 0001, Hai-Tao Zheng 0002, Stefan Wermter, Tat-Seng Chua, Maosong Sun 0001 |
AAAI | 11 |
| 2023 | Weakly Supervised Vision-and-Language Pre-training with Relative RepresentationsabstractWeakly supervised vision-and-language pretraining (WVLP), which learns cross-modal representations with limited cross-modal supervision, has been shown to effectively reduce the data cost of pre-training while maintaining decent performance on downstream tasks.However, current WVLP methods use only local descriptions of images, i.e., object tags, as cross-modal anchors to construct weaklyaligned image-text pairs for pre-training.This affects the data quality and thus the effectiveness of pre-training.In this paper, we propose to directly take a small number of aligned image-text pairs as anchors, and represent each unaligned image and text by its similarities to these anchors, i.e., relative representations.We build a WVLP framework based on the relative representations, namely RELIT 1 , which collects high-quality weakly-aligned imagetext pairs from large-scale image-only and text-only data for pre-training through relative representation-based retrieval and generation.Experiments on four downstream tasks show that RELIT achieves new state-of-the-art results under the weakly supervised setting 2 . Chi Chen 0005, Peng Li 0030, Maosong Sun 0001, Yang Liu 0005 |
ACL (1) | 3 |
| 2023 | Decoder Tuning: Efficient Language Understanding as DecodingabstractWith the evergrowing sizes of pre-trained models (PTMs), it has been an emerging practice to only provide the inference APIs for users, namely model-as-a-service (MaaS) setting.To adapt PTMs with model parameters frozen, most current approaches focus on the input side, seeking for powerful prompts to stimulate models for correct answers.However, we argue that input-side adaptation could be arduous due to the lack of gradient signals and they usually require thousands of API queries, resulting in high computation and time costs.In light of this, we present Decoder Tuning (DecT), which in contrast optimizes task-specific decoder networks on the output side.Specifically, DecT first extracts prompt-stimulated output scores for initial predictions.On top of that, we train an additional decoder network on the output representations to incorporate posterior data knowledge.By gradientbased optimization, DecT can be trained within several seconds and requires only one PTM query per sample.Empirically, we conduct extensive natural language understanding experiments and show that DecT significantly outperforms state-of-the-art algorithms with a 200× speed-up.Our codes are available at https://github.com/thunlp/DecT. Ganqu Cui, Ning Ding 0002, Longtao Huang, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 6 |
| 2023 | Won't Get Fooled Again: Answering Questions with False PremisesabstractPre-trained language models (PLMs) have shown unprecedented potential in various fields, especially as the backbones for questionanswering (QA) systems.However, they tend to be easily deceived by tricky questions such as "How many eyes does the sun have?".Such frailties of PLMs often allude to the lack of knowledge within them.In this paper, we find that the PLMs already possess the knowledge required to rebut such questions, and the key is how to activate the knowledge.To systematize this observation, we investigate the PLMs' responses to one kind of tricky questions, i.e., the false premises questions (FPQs).We annotate a FalseQA dataset containing 2365 human-written FPQs, with the corresponding explanations for the false premises and the revised true premise questions.Using FalseQA, we discover that PLMs are capable of discriminating FPQs by fine-tuning on moderate numbers (e.g., 256) of examples.PLMs also generate reasonable explanations for the false premise, which serve as rebuttals.Further replaying a few general questions during training allows PLMs to excel on FPQs and general questions simultaneously.Our work suggests that once the rebuttal ability is stimulated, knowledge inside the PLMs can be effectively utilized to handle FPQs, which incentivizes the research on PLM-based QA systems.The FalseQA dataset and code are available at https://github.com/thunlp/FalseQA. Shengding Hu, Xingyi Cheng, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 6 |
| 2023 | An Extensible Plug-and-Play Method for Multi-Aspect Controllable Text GenerationabstractRecently, multi-aspect controllable text generation that controls the generated text in multiple aspects (e.g., sentiment, topic, and keywords) has attracted increasing attention.Although methods based on parameter efficient tuning like prefix-tuning could achieve multi-aspect controlling in a plug-and-play way, the mutual interference of multiple prefixes leads to significant degeneration of constraints and limits their extensibility to training-time unseen aspect combinations.In this work, we provide a theoretical lower bound for the interference and empirically found that the interference grows with the number of layers where prefixes are inserted.Based on these analyses, we propose using trainable gates to normalize the intervention of prefixes to restrain the growing interference.As a result, controlling training-time unseen combinations of aspects can be realized by simply concatenating corresponding plugins such that new constraints can be extended at a lower cost.In addition, we propose a unified way to process both categorical and free-form constraints.Experiments on text generation and machine translation demonstrate the superiority of our approach over baselines on constraint accuracy, text quality, and extensibility. 1 Xuancheng Huang, Peng Li 0030, Maosong Sun 0001, Yang Liu 0005 |
ACL (1) | 5 |
| 2023 | WebCPM: Interactive Web Search for Chinese Long-form Question AnsweringabstractYujia Qin, Zihan Cai, Dian Jin, Lan Yan, Shihao Liang, Kunlun Zhu, Yankai Lin, Xu Han, Ning Ding, Huadong Wang, Ruobing Xie, Fanchao Qi, Zhiyuan Liu, Maosong Sun, Jie Zhou. Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2023. Yujia Qin, Zihan Cai, Lan Yan, Shihao Liang, Kunlun Zhu, Yankai Lin 0001, Xu Han 0007, Ning Ding 0002, Ruobing Xie, Fanchao Qi, Zhiyuan Liu 0001, Maosong Sun 0001, Jie Zhou 0024 |
ACL (1) | 14 |
| 2023 | READIN: A Chinese Multi-Task Benchmark with Realistic and Diverse Input NoisesabstractFor many real-world applications, the usergenerated inputs usually contain various noises due to speech recognition errors caused by linguistic variations 1 or typographical errors (typos).Thus, it is crucial to test model performance on data with realistic input noises to ensure robustness and fairness.However, little study has been done to construct such benchmarks for Chinese, where various languagespecific input noises happen in the real world.In order to fill this important gap, we construct READIN: a Chinese multi-task benchmark with REalistic And Diverse Input Noises.READIN contains four diverse tasks and requests annotators to re-enter the original test data with two commonly used Chinese input methods: Pinyin input and speech input.We designed our annotation pipeline to maximize diversity, for example by instructing the annotators to use diverse input method editors (IMEs) for keyboard noises and recruiting speakers from diverse dialectical groups for speech noises.We experiment with a series of strong pretrained language models as well as robust training methods, we find that these models often suffer significant performance drops on READIN even with robustness methods like data augmentation.As the first large-scale attempt in creating a benchmark with noises geared towards user-generated inputs, we believe that READIN serves as an important complement to existing Chinese NLP benchmarks.The source code and dataset can be obtained from https://github.com/ thunlp/READIN. Chenglei Si, Zhengyan Zhang, Yingfa Chen, Xiaozhi Wang, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 6 |
| 2023 | Plug-and-Play Document Modules for Pre-trained ModelsabstractChaojun Xiao, Zhengyan Zhang, Xu Han, Chi-Min Chan, Yankai Lin, Zhiyuan Liu, Xiangyang Li, Zhonghua Li, Zhao Cao, Maosong Sun. Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2023. Chaojun Xiao, Zhengyan Zhang, Xu Han 0007, Chi-Min Chan, Yankai Lin 0001, Zhiyuan Liu 0001, Zhao Cao, Maosong Sun 0001 |
ACL (1) | 10 |
| 2023 | Continual Knowledge Distillation for Neural Machine TranslationabstractWhile many parallel corpora are not publicly accessible for data copyright, data privacy and competitive differentiation reasons, trained translation models are increasingly available on open platforms.In this work, we propose a method called continual knowledge distillation to take advantage of existing translation models to improve one model of interest.The basic idea is to sequentially transfer knowledge from each trained model to the distilled model.Extensive experiments on Chinese-English and German-English datasets show that our method achieves significant and consistent improvements over strong baselines under both homogeneous and heterogeneous trained model settings and is robust to malicious models. Yuanchi Zhang, Peng Li 0030, Maosong Sun 0001, Yang Liu 0005 |
ACL (1) | 3 |
| 2023 | Plug-and-Play Knowledge Injection for Pre-trained Language ModelsabstractZhengyan Zhang, Zhiyuan Zeng, Yankai Lin, Huadong Wang, Deming Ye, Chaojun Xiao, Xu Han, Zhiyuan Liu, Peng Li, Maosong Sun, Jie Zhou. Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2023. Zhengyan Zhang, Yankai Lin 0001, Deming Ye, Chaojun Xiao, Xu Han 0007, Zhiyuan Liu 0001, Peng Li 0030, Maosong Sun 0001, Jie Zhou 0016 |
ACL (1) | 10 |
| 2023 | MUSER: A Multi-View Similar Case Retrieval DatasetabstractSimilar case retrieval (SCR) is a representative legal AI application that plays a pivotal role in promoting judicial fairness. However, existing SCR datasets only focus on the fact description section when judging the similarity between cases, ignoring other valuable sections (e.g., the court's opinion) that can provide insightful reasoning process behind. Furthermore, the case similarities are typically measured solely by the textual semantics of the fact descriptions, which may fail to capture the full complexity of legal cases from the perspective of legal knowledge. In this work, we present MUSER, a similar case retrieval dataset based on multi-view similarity measurement and comprehensive legal element with sentence-level legal element annotations. Specifically, we select three perspectives (legal fact, dispute focus, and law statutory) and build a comprehensive and structured label schema of legal elements for each of them, to enable accurate and knowledgeable evaluation of case similarities. The constructed dataset originates from Chinese civil cases and contains 100 query cases and 4,024 candidate cases. We implement several text classification algorithms for legal element prediction and various retrieval methods for retrieving similar cases on MUSER. The experimental results indicate that incorporating legal elements can benefit the performance of SCR models, but further efforts are still required to address the remaining challenges posed by MUSER. The source code and dataset are released at https://github.com/THUlawtech/MUSER. Qingquan Li 0003, Yiran Hu, Chaojun Xiao, Zhiyuan Liu 0001, Maosong Sun 0001, Weixing Shen |
CIKM | 6 |
| 2023 | Enhancing Chat Language Models by Scaling High-quality Instructional ConversationsabstractFine-tuning on instruction data has been widely validated as an effective practice for implementing chat language models like ChatGPT.Scaling the diversity and quality of such data, although straightforward, stands a great chance of leading to improved performance.This paper aims to push the upper bound of opensource models further.We first provide a systematically designed, diverse, informative, large-scale dataset of instructional conversations, UltraChat, which does not involve human queries.Our objective is to capture the breadth of interactions between a human user and an AI assistant and employs a comprehensive framework to generate multi-turn conversation iteratively.UltraChat contains 1.5 million high-quality multi-turn dialogues and covers a wide range of topics and instructions.Our statistical analysis of UltraChat reveals its superiority in various key metrics, including scale, average length, diversity, coherence, etc., solidifying its position as a leading opensource dataset.Building upon UltraChat, we fine-tune a LLaMA model to create a powerful conversational model, UltraLM.Our evaluations indicate that UltraLM consistently outperforms other open-source models, including WizardLM and Vicuna, the previously recognized state-of-the-art open-source models. Ning Ding 0002, Yulin Chen 0001, Bokai Xu, Yujia Qin, Shengding Hu, Zhiyuan Liu 0001, Maosong Sun 0001, Bowen Zhou 0002 |
EMNLP | 7 |
| 2023 | Sparse Low-rank Adaptation of Pre-trained Language ModelsabstractFine-tuning pre-trained large language models in a parameter-efficient manner is widely studied for its effectiveness and efficiency.The popular method of low-rank adaptation (LoRA) offers a notable approach, hypothesizing that the adaptation process is intrinsically low-dimensional.Although LoRA has demonstrated commendable performance, it is implemented with a fixed and unalterable intrinsic rank that might not always be the ideal choice.Recognizing the need for more flexible adaptation, we extend the methodology of LoRA to an innovative approach we call sparse low-rank adaptation (SoRA) that enables dynamic adjustments to the intrinsic rank during the adaptation process.We achieve this through the incorporation of a gate unit optimized with proximal gradient method in the training stage, controlling the cardinality of rank under the sparsity of the gate.In the subsequent inference stage, we eliminate the parameter blocks corresponding to the zeroed-out ranks, to reduce each SoRA module back to a concise yet rankoptimal LoRA.Our approach strengthens the representation power of LoRA by initializing it with a higher rank, while efficiently taming a temporarily increased number of parameters via updating in a sparse way.We further introduce a sparsifying scheduler for SoRA, aiming to examine the impact of the number of nonzero parameters on the model's memorization and generalization.Our experimental results demonstrate that SoRA can outperform other baselines even with 70% retained parameters and 70% training time. Ning Ding 0002, Xingtai Lv, Qiaosen Wang, Yulin Chen 0001, Bowen Zhou 0002, Zhiyuan Liu 0001, Maosong Sun 0001 |
EMNLP | 7 |
| 2023 | Learn and Consolidate: Continual Adaptation for Zero-Shot and Multilingual Neural Machine TranslationabstractAlthough existing multilingual neural machine translation (MNMT) models have demonstrated remarkable performance to handle multiple translation directions in a single model and achieved zero-shot translation between language pairs unseen in training, they still suffer from relatively poor translation qualities for some language pairs.A practical scenario is that how to continually update MNMT models for both supervised and zero-shot translations when limited new data arrives.To this end, we propose a two-stage approach that encourages original models to acquire language-agnostic multilingual representations from new data, and preserves the model architecture without introducing parameters.Experimental results and further analysis demonstrate that our method can efficiently improve performance of existing MNMT models in translation directions where they are initially weak, and mitigates the degeneration in the original well-performing translation directions, offering flexibility in the real-world scenario. Peng Li 0030, Junpeng Liu 0002, Maosong Sun 0001, Yang Liu 0005 |
EMNLP | 4 |
| 2023 | Exploring the Impact of Model Scaling on Parameter-Efficient TuningabstractYusheng Su, Chi-Min Chan, Jiali Cheng, Yujia Qin, Yankai Lin, Shengding Hu, Zonghan Yang, Ning Ding, Xingzhi Sun, Guotong Xie, Zhiyuan Liu, Maosong Sun. Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing. 2023. Yusheng Su, Chi-Min Chan, Jiali Cheng, Yujia Qin, Yankai Lin 0001, Shengding Hu, Zonghan Yang, Ning Ding 0002, Xingzhi Sun 0002, Guo Tong Xie, Zhiyuan Liu 0001, Maosong Sun 0001 |
EMNLP | 12 |
| 2023 | Beat LLMs at Their Own Game: Zero-Shot LLM-Generated Text Detection via Querying ChatGPTabstractBiru Zhu, Lifan Yuan, Ganqu Cui, Yangyi Chen, Chong Fu, Bingxiang He, Yangdong Deng, Zhiyuan Liu, Maosong Sun, Ming Gu. Proceedings of the 2023 Conference on Empirical Methods in Natural Language Processing. 2023. Biru Zhu, Lifan Yuan, Ganqu Cui, Yangyi Chen, Bingxiang He, Yangdong Deng, Zhiyuan Liu 0001, Maosong Sun 0001, Ming Gu 0001 |
EMNLP | 9 |
| 2023 | Chord-Conditioned Melody Harmonization With Controllable HarmonicityabstractMelody harmonization has long been closely associated with chorales composed by Johann Sebastian Bach. Previous works rarely emphasised chorale generation conditioned on chord progressions, and there has been a lack of focus on assistive compositional tools. In this paper, we first designed a music representation that encoded chord symbols for chord conditioning, and then proposed DeepChoir1, a melody harmonization system that can generate a four-part chorale for a given melody conditioned on a chord progression. With controllable harmonicity, users can control the extent of harmonicity for generated chorales. Experimental results reveal the effectiveness of the music representation and the controllability of DeepChoir. Shangda Wu, Maosong Sun 0001 |
ICASSP | 3 |
| 2023 | C-Eval: A Multi-Level Multi-Discipline Chinese Evaluation Suite for Foundation ModelsabstractNew NLP benchmarks are urgently needed to align with the rapid development of large language models (LLMs). We present C-Eval, the first comprehensive Chinese evaluation suite designed to assess advanced knowledge and reasoning abilities of foundation models in a Chinese context. C-Eval comprises multiple-choice questions across four difficulty levels: middle school, high school, college, and professional. The questions span 52 diverse disciplines, ranging from humanities to science and engineering. C-Eval is accompanied by C-Eval Hard, a subset of very challenging subjects in C-Eval that requires advanced reasoning abilities to solve. We conduct a comprehensive evaluation of the most advanced LLMs on C-Eval, including both English- and Chinese-oriented models. Results indicate that only GPT-4 could achieve an average accuracy of over 60%, suggesting that there is still significant room for improvement for current LLMs. We anticipate C-Eval will help analyze important strengths and shortcomings of foundation models, and foster their development and growth for Chinese users. Yuzhen Huang 0002, Yuzhuo Bai, Junlei Zhang, Jinghan Zhang 0006, Tangjun Su, Junteng Liu, Chuancheng Lv, Jiayi Lei, Maosong Sun 0001, Junxian He |
NeurIPS | 12 |
| 2023 | H3T: Efficient Integration of Memory Optimization and Parallelism for Large-scale Transformer TrainingabstractIn recent years, big models based on Transformers have achieved state-of-the-art performance on many artificial intelligence (AI) tasks.
Despite the success of these Transformer-based models, their huge parameter size poses a serious challenge to their training, both from the storage and computation perspectives.
To this end, memory optimization (e.g., rematerialization and offloading) and parallelism (e.g., data parallelism and model parallelism) are widely explored to make training Transformers more efficient.
In this paper, we propose a framework to automatically find an efficient integration of memory optimization and parallelism for High-Throughput Transformer Training (named H3T), which is rarely considered by existing efforts for training big Transformer-based models.
Specifically, we design search algorithms to combine appropriate memory optimization strategies and parallelism schemes to achieve a balance between memory overhead and training efficiency.
We implement H3T based on an open-source toolkit BMTrain and then use H3T to train the Transformers of different sizes to evaluate the efficiency of H3T.
The experimental results show that H3T outperforms the most popular deep learning (DL) toolkit Megatron-DeepSpeed by $1.2\times \sim 4.3\times$ training speed while reducing $34.6\% \sim 80.5\%$ of memory overhead.
Moreover, H3T can use only 64 NVIDIA A100 GPUs to train GPT-3-175B, which is very difficult for existing DL toolkits. The source code is available at https://github.com/OpenBMB/BMTrain/tree/h3t. Xu Han 0007, Weilin Zhao, Guoyang Zeng, Zhiyuan Liu 0001, Maosong Sun 0001 |
NeurIPS | 6 |
| 2023 | Revisiting Out-of-distribution Robustness in NLP: Benchmarks, Analysis, and LLMs EvaluationsabstractThis paper reexamines the research on out-of-distribution (OOD) robustness in the field of NLP. We find that the distribution shift settings in previous studies commonly lack adequate challenges, hindering the accurate evaluation of OOD robustness. To address these issues, we propose a benchmark construction protocol that ensures clear differentiation and challenging distribution shifts. Then we introduceBOSS, a Benchmark suite for Out-of-distribution robustneSS evaluation covering 5 tasks and 20 datasets. Based on BOSS, we conduct a series of experiments on pretrained language models for analysis and evaluation of OOD robustness. First, for vanilla fine-tuning, we examine the relationship between in-distribution (ID) and OOD performance. We identify three typical types that unveil the inner learningmechanism, which could potentially facilitate the forecasting of OOD robustness, correlating with the advancements on ID datasets. Then, we evaluate 5 classic methods on BOSS and find that, despite exhibiting some effectiveness in specific cases, they do not offer significant improvement compared to vanilla fine-tuning. Further, we evaluate 5 LLMs with various adaptation paradigms and find that when sufficient ID data is available, fine-tuning domain-specific models outperform LLMs on ID examples significantly. However, in the case of OOD instances, prioritizing LLMs with in-context learning yields better results. We identify that both fine-tuned small models and LLMs face challenges in effectively addressing downstream tasks. The code is public at https://github.com/lifan-yuan/OOD_NLP. Lifan Yuan, Yangyi Chen, Ganqu Cui, Hongcheng Gao, Fangyuan Zou, Xingyi Cheng, Heng Ji 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
NeurIPS | 9 |
| 2023 | Improve the Diversity and Novelty for Open-Ended Neural Text Generation via Inverse Probability Weighting
Xinran Zhang 0003, Maosong Sun 0001, Jiafeng Liu |
NLPCC (1) | 2 |
| 2023 | Sub-Character Tokenization for Chinese Pretrained Language ModelsabstractAbstract Tokenization is fundamental to pretrained language models (PLMs). Existing tokenization methods for Chinese PLMs typically treat each character as an indivisible token. However, they ignore the unique feature of the Chinese writing system where additional linguistic information exists below the character level, i.e., at the sub-character level. To utilize such information, we propose sub-character (SubChar for short) tokenization. Specifically, we first encode the input text by converting each Chinese character into a short sequence based on its glyph or pronunciation, and then construct the vocabulary based on the encoded text with sub-word segmentation. Experimental results show that SubChar tokenizers have two main advantages over existing tokenizers: 1) They can tokenize inputs into much shorter sequences, thus improving the computational efficiency. 2) Pronunciation-based SubChar tokenizers can encode Chinese homophones into the same transliteration sequences and produce the same tokenization output, hence being robust to homophone typos. At the same time, models trained with SubChar tokenizers perform competitively on downstream tasks. We release our code and models at https://github.com/thunlp/SubCharTokenization to facilitate future work. Chenglei Si, Zhengyan Zhang, Yingfa Chen, Fanchao Qi, Xiaozhi Wang, Zhiyuan Liu 0001, Yasheng Wang, Qun Liu 0001, Maosong Sun 0001 |
Trans. Assoc. Comput. Linguistics | 9 |
| 2023 | Removing Backdoors in Pre-trained Models by Regularized Continual Pre-trainingabstractAbstract Recent research has revealed that pre-trained models (PTMs) are vulnerable to backdoor attacks before the fine-tuning stage. The attackers can implant transferable task-agnostic backdoors in PTMs, and control model outputs on any downstream task, which poses severe security threats to all downstream applications. Existing backdoor-removal defenses focus on task-specific classification models and they are not suitable for defending PTMs against task-agnostic backdoor attacks. To this end, we propose the first task-agnostic backdoor removal method for PTMs. Based on the selective activation phenomenon in backdoored PTMs, we design a simple and effective backdoor eraser, which continually pre-trains the backdoored PTMs with a regularization term in an end-to-end approach. The regularization term removes backdoor functionalities from PTMs while the continual pre-training maintains the normal functionalities of PTMs. We conduct extensive experiments on pre-trained models across different modalities and architectures. The experimental results show that our method can effectively remove backdoors inside PTMs and preserve benign functionalities of PTMs with a few downstream-task-irrelevant auxiliary data, e.g., unlabeled plain texts. The average attack success rate on three downstream datasets is reduced from 99.88% to 8.10% after our defense on the backdoored BERT. The codes are publicly available at https://github.com/thunlp/RECIPE. Biru Zhu, Ganqu Cui, Yangyi Chen, Yujia Qin, Lifan Yuan, Yangdong Deng, Zhiyuan Liu 0001, Maosong Sun 0001, Ming Gu 0001 |
Trans. Assoc. Comput. Linguistics | 9 |
| 2023 | Full-Scale Information Diffusion Prediction With Reinforced Recurrent NetworksabstractInformation diffusion prediction is an important task, which studies how information items spread among users. With the success of deep learning techniques, recurrent neural networks (RNNs) have shown their powerful capability in modeling information diffusion as sequential data. However, previous works focused on either microscopic diffusion prediction, which aims at guessing who will be the next influenced user at what time, or macroscopic diffusion prediction, which estimates the total numbers of influenced users during the diffusion process. To the best of our knowledge, few attempts have been made to suggest a unified model for both microscopic and macroscopic scales. In this article, we propose a novel full-scale diffusion prediction model based on reinforcement learning (RL). RL incorporates the macroscopic diffusion size information into the RNN-based microscopic diffusion model by addressing the nondifferentiable problem. We also employ an effective structural context extraction strategy to utilize the underlying social graph information. Experimental results show that our proposed model outperforms state-of-the-art baseline models on both microscopic and macroscopic diffusion predictions on three real-world datasets. Cheng Yang 0002, Hao Wang 0139, Jian Tang 0005, Chuan Shi 0001, Maosong Sun 0001, Ganqu Cui, Zhiyuan Liu 0001 |
IEEE Trans. Neural Networks Learn. Syst. | 5 |
| 2022 | Fully Hyperbolic Neural NetworksabstractWeize Chen, Xu Han, Yankai Lin, Hexu Zhao, Zhiyuan Liu, Peng Li, Maosong Sun, Jie Zhou. Proceedings of the 60th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2022. Weize Chen, Xu Han 0007, Yankai Lin 0001, Hexu Zhao, Zhiyuan Liu 0001, Peng Li 0030, Maosong Sun 0001, Jie Zhou 0016 |
ACL (1) | 7 |
| 2022 | Cross-Lingual Contrastive Learning for Fine-Grained Entity Typing for Low-Resource LanguagesabstractXu Han, Yuqi Luo, Weize Chen, Zhiyuan Liu, Maosong Sun, Zhou Botong, Hao Fei, Suncong Zheng. Proceedings of the 60th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2022. Xu Han 0007, Yuqi Luo, Weize Chen, Zhiyuan Liu 0001, Maosong Sun 0001, Botong Zhou, Suncong Zheng |
ACL (1) | 5 |
| 2022 | Knowledgeable Prompt-tuning: Incorporating Knowledge into Prompt Verbalizer for Text ClassificationabstractShengding Hu, Ning Ding, Huadong Wang, Zhiyuan Liu, Jingang Wang, Juanzi Li, Wei Wu, Maosong Sun. Proceedings of the 60th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2022. Shengding Hu, Ning Ding 0002, Zhiyuan Liu 0001, Jingang Wang, Juan-Zi Li, Wei Wu 0014, Maosong Sun 0001 |
ACL (1) | 8 |
| 2022 | TopWORDS-Seg: Simultaneous Text Segmentation and Word Discovery for Open-Domain Chinese Texts via Bayesian InferenceabstractProcessing open-domain Chinese texts has been a critical bottleneck in computational linguistics for decades, partially because text segmentation and word discovery often entangle with each other in this challenging scenario.No existing methods yet can achieve effective text segmentation and word discovery simultaneously in open domain.This study fills in this gap by proposing a novel method called TopWORDS-Seg based on Bayesian inference, which enjoys robust performance and transparent interpretation when no training corpus and domain vocabulary are available.Advantages of TopWORDS-Seg are demonstrated by a series of experimental studies. Changzai Pan, Maosong Sun 0001 |
ACL (1) | 2 |
| 2022 | QuoteR: A Benchmark of Quote Recommendation for WritingabstractIt is very common to use quotations (quotes) to make our writings more elegant or convincing.To help people find appropriate quotes efficiently, the task of quote recommendation is presented, aiming to recommend quotes that fit the current context of writing.There have been various quote recommendation approaches, but they are evaluated on different unpublished datasets.To facilitate the research on this task, we build a large and fully open quote recommendation dataset called QuoteR, which comprises three parts including English, standard Chinese and classical Chinese.Any part of it is larger than previous unpublished counterparts.We conduct an extensive evaluation of existing quote recommendation methods on QuoteR.Furthermore, we propose a new quote recommendation model that significantly outperforms previous methods on all three parts of QuoteR.All the code and data of this paper can be obtained at https://github.com/ thunlp/QuoteR. Fanchao Qi, Yanhui Yang, Jing Yi, Zhili Cheng, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 6 |
| 2022 | Packed Levitated Marker for Entity and Relation ExtractionabstractRecent entity and relation extraction works focus on investigating how to obtain a better span representation from the pre-trained encoder. However, a major limitation of existing works is that they ignore the interrelation between spans (pairs). In this work, we propose a novel span representation approach, named Packed Levitated Markers (PL-Marker), to consider the interrelation between the spans (pairs) by strategically packing the markers in the encoder. In particular, we propose a neighborhood-oriented packing strategy, which considers the neighbor spans integrally to better model the entity boundary information. Furthermore, for those more complicated span pair classification tasks, we design a subject-oriented packing strategy, which packs each subject and all its objects to model the interrelation between the same-subject span pairs. The experimental results show that, with the enhanced marker feature, our model advances baselines on six NER benchmarks, and obtains a 4.1%-4.3% strict relation F1 improvement with higher speed over previous state-of-the-art models on ACE04 and ACE05. Our code and models are publicly available at https://github.com/thunlp/PL-Marker Deming Ye, Yankai Lin 0001, Peng Li 0030, Maosong Sun 0001 |
ACL (1) | 4 |
| 2022 | Pass off Fish Eyes for Pearls: Attacking Model Selection of Pre-trained ModelsabstractBiru Zhu, Yujia Qin, Fanchao Qi, Yangdong Deng, Zhiyuan Liu, Maosong Sun, Ming Gu. Proceedings of the 60th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2022. Biru Zhu, Yujia Qin, Fanchao Qi, Yangdong Deng, Zhiyuan Liu 0001, Maosong Sun 0001, Ming Gu 0001 |
ACL (1) | 6 |
| 2022 | Automatic Label Sequence Generation for Prompting Sequence-to-sequence ModelsabstractPrompting, which casts downstream applications as language modeling tasks, has shown to be sample efficient compared to standard fine-tuning with pre-trained models. However, one pitfall of prompting is the need of manually-designed patterns, whose outcome can be unintuitive and requires large validation sets to tune. To tackle the challenge, we propose AutoSeq, a fully automatic prompting method: (1) We adopt natural language prompts on sequence-to-sequence models, enabling free-form generation and larger label search space; (2) We propose label sequences – phrases with indefinite lengths to verbalize the labels – which eliminate the need of manual templates and are more expressive than single label words; (3) We use beam search to automatically generate a large amount of label sequence candidates and propose contrastive re-ranking to get the best combinations. AutoSeq significantly outperforms other no-manual-design methods, such as soft prompt tuning, adapter tuning, and automatic search on single label words; the generated label sequences are even better than curated manual ones on a variety of tasks. Our method reveals the potential of sequence-to-sequence models in few-shot learning and sheds light on a path to generic and automatic prompting. The source code of this paper can be obtained from https://github.com/thunlp/Seq2Seq-Prompt. Zichun Yu, Tianyu Gao 0001, Zhengyan Zhang, Yankai Lin 0001, Zhiyuan Liu 0001, Maosong Sun 0001, Jie Zhou 0016 |
COLING | 6 |
| 2022 | Fine-Grained Scene Graph Generation with Data Transfer
Yuan Yao 0013, Wei Ji 0008, Zhiyuan Liu 0001, Maosong Sun 0001, Tat-Seng Chua |
ECCV (27) | 6 |
| 2022 | Why Should Adversarial Perturbations be Imperceptible? Rethink the Research Paradigm in Adversarial NLPabstractTextual adversarial samples play important roles in multiple subfields of NLP research, including security, evaluation, explainability, and data augmentation.However, most work mixes all these roles, obscuring the problem definitions and research goals of the security role that aims to reveal the practical concerns of NLP models.In this paper, we rethink the research paradigm of textual adversarial samples in security scenarios.We discuss the deficiencies in previous work and propose our suggestions that the research on the Security-oriented adversarial NLP (SoadNLP) should: (1) evaluate their methods on security tasks to demonstrate the real-world concerns; (2) consider realworld attackers' goals, instead of developing impractical methods.To this end, we first collect, process, and release a security datasets collection Advbench.Then, we reformalize the task and adjust the emphasis on different goals in SoadNLP.Next, we propose a simple method based on heuristic rules that can easily fulfill the actual adversarial goals to simulate real-world attack methods.We conduct experiments on both the attack and the defense sides on Advbench.Experimental results show that our method has higher practical value, indicating that the research paradigm in SoadNLP may start from our new benchmark.All the code and data of Advbench can be obtained at https: //github.com/thunlp/Advbench. Yangyi Chen, Hongcheng Gao, Ganqu Cui, Fanchao Qi, Longtao Huang, Zhiyuan Liu 0001, Maosong Sun 0001 |
EMNLP | 7 |
| 2022 | End-to-End Unsupervised Vision-and-Language Pre-training with Referring Expression MatchingabstractRecently there has been an emerging interest in unsupervised vision-and-language pre-training (VLP) that learns multimodal representations without parallel image-caption data.These pioneering works significantly reduce the cost of VLP on data collection and achieve promising results compared to supervised VLP.However, existing unsupervised VLP methods take as input pre-extracted region-based visual features from external object detectors, which both limits flexibility and reduces computational efficiency.In this paper, we explore end-to-end unsupervised VLP with a vision encoder to directly encode images.The vision encoder is pre-trained on image-only data and jointly optimized during multimodal pre-training.To further enhance the learned cross-modal features, we propose a novel pre-training task that predicts which patches contain an object referred to in natural language from the encoded visual features.Extensive experiments on four visionand-language tasks show that our approach outperforms previous unsupervised VLP methods and obtains new state-of-the-art results 1 . Chi Chen 0005, Peng Li 0030, Maosong Sun 0001, Yang Liu 0005 |
EMNLP | 3 |
| 2022 | Textual Backdoor Attacks Can Be More Harmful via Two Simple TricksabstractBackdoor attacks are a kind of emergent security threat in deep learning.After being injected with a backdoor, a deep neural model will behave normally on standard inputs but give adversary-specified predictions once the input contains specific backdoor triggers.In this paper, we find two simple tricks that can make existing textual backdoor attacks much more harmful.The first trick is to add an extra training task to distinguish poisoned and clean data during the training of the victim model, and the second one is to use all the clean training data rather than remove the original clean data corresponding to the poisoned data.These two tricks are universally applicable to different attack models.We conduct experiments in three tough situations including clean data fine-tuning, low-poisoningrate, and label-consistent attacks.Experimental results show that the two tricks can significantly improve attack performance.This paper exhibits the great potential harmfulness of backdoor attacks.All the code and data can be obtained at https://github.com/ thunlp/StyleAttack. Yangyi Chen, Fanchao Qi, Hongcheng Gao, Zhiyuan Liu 0001, Maosong Sun 0001 |
EMNLP | 5 |
| 2022 | Evade the Trap of Mediocrity: Promoting Diversity and Novelty in Text Generation via Concentrating AttentionabstractRecently, powerful Transformer architectures have proven superior in generating high-quality sentences.Nevertheless, these models tend to produce dull high-frequency phrases, severely hurting the diversity and novelty of generated text.In this work, we dig into the intrinsic mechanism of this problem and found that sparser attention values in Transformer could improve diversity.To understand such a phenomenon, we first conduct both empirical and theoretical analysis and then attribute it to representation degeneration caused by the attentive mixture of the hidden states during training.We term this process the Trap of Mediocrity.To escape from such a trap, we introduce a novel attention regularization loss to control the sharpness of the attention distribution, which is transparent to model structures and can be easily implemented within 20 lines of python code.We prove that this method could be mathematically regarded as learning a Bayesian approximation of posterior attention.Experiments show that our method improved the diversity and novelty of the generated text while maintaining comparable quality on a variety of conditional and unconditional generation tasks. Wenhao Li 0003, Xiaoyuan Yi, Jinyi Hu, Maosong Sun 0001, Xing Xie 0001 |
EMNLP | 4 |
| 2022 | Exploring Mode Connectivity for Pre-trained Language ModelsabstractRecent years have witnessed the prevalent application of pre-trained language models (PLMs) in NLP.From the perspective of parameter space, PLMs provide generic initialization, starting from which high-performance minima could be found.Although plenty of works have studied how to effectively and efficiently adapt PLMs to high-performance minima, little is known about the connection of various minima reached under different adaptation configurations.In this paper, we investigate the geometric connections of different minima through the lens of mode connectivity, which measures whether two minima can be connected with a low-loss path.We conduct empirical analyses to investigate three questions: (1) how could hyperparameters, specific tuning methods, and training data affect PLM's mode connectivity?(2) How does mode connectivity change during pretraining?(3) How does the PLM's task knowledge change along the path connecting two minima?In general, exploring the mode connectivity of PLMs conduces to understanding the geometric connection of different minima, which may help us fathom the inner workings of PLM downstream adaptation.The codes are publicly available at https://github.com/ thunlp/Mode-Connectivity-PLM. Yujia Qin, Cheng Qian 0008, Jing Yi, Weize Chen, Yankai Lin 0001, Xu Han 0007, Zhiyuan Liu 0001, Maosong Sun 0001, Jie Zhou 0016 |
EMNLP | 8 |
| 2022 | A Template-based Method for Constrained Neural Machine TranslationabstractMachine translation systems are expected to cope with various types of constraints in many practical scenarios.While neural machine translation (NMT) has achieved strong performance in unconstrained cases, it is non-trivial to impose pre-specified constraints into the translation process of NMT models.Although many approaches have been proposed to address this issue, most existing methods can not satisfy the following three desiderata at the same time: (1) high translation quality, (2) high match accuracy, and (3) low latency.In this work, we propose a template-based method that can yield results with high translation quality and match accuracy and the inference speed of our method is comparable with unconstrained NMT models.Our basic idea is to rearrange the generation of constrained and unconstrained tokens through a template.Our method does not require any changes in the model architecture and the decoding algorithm.Experimental results show that the proposed template-based approach can outperform several representative baselines in both lexically and structurally constrained translation tasks. Shuo Wang 0013, Peng Li 0030, Zhixing Tan, Zhaopeng Tu, Maosong Sun 0001, Yang Liu 0005 |
EMNLP | 5 |
| 2022 | PEVL: Position-enhanced Pre-training and Prompt Tuning for Vision-language ModelsabstractVision-language pre-training (VLP) has shown impressive performance on a wide range of cross-modal tasks, where VLP models without reliance on object detectors are becoming the mainstream due to their superior computation efficiency and competitive performance.However, the removal of object detectors also deprives the capability of VLP models in explicit object modeling, which is essential to various position-sensitive vision-language (VL) tasks, such as referring expression comprehension and visual commonsense reasoning.To address the challenge, we introduce PEVL that enhances the pre-training and prompt tuning of VLP models with explicit object position modeling.Specifically, PEVL reformulates discretized object positions and language in a unified language modeling framework, which facilitates explicit VL alignment during pretraining, and also enables flexible prompt tuning for various downstream tasks.We show that PEVL enables state-of-the-art performance of detector-free VLP models on position-sensitive tasks such as referring expression comprehension and phrase grounding, and also improves the performance on position-insensitive tasks with grounded inputs.We make the data and code for this paper publicly available at https://github.com/thunlp/PEVL. Yuan Yao 0013, Wei Ji 0008, Zhiyuan Liu 0001, Tat-Seng Chua, Maosong Sun 0001 |
EMNLP | 7 |
| 2022 | Fuse It More Deeply! A Variational Transformer with Layer-Wise Latent Variable Inference for Text GenerationabstractJinyi Hu, Xiaoyuan Yi, Wenhao Li, Maosong Sun, Xing Xie. Proceedings of the 2022 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies. 2022. Jinyi Hu, Xiaoyuan Yi, Wenhao Li 0003, Maosong Sun 0001, Xing Xie 0001 |
NAACL-HLT | 4 |
| 2022 | Knowledge Inheritance for Pre-trained Language ModelsabstractYujia Qin, Yankai Lin, Jing Yi, Jiajie Zhang, Xu Han, Zhengyan Zhang, Yusheng Su, Zhiyuan Liu, Peng Li, Maosong Sun, Jie Zhou. Proceedings of the 2022 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies. 2022. Yujia Qin, Yankai Lin 0001, Jing Yi, Xu Han 0007, Zhengyan Zhang, Yusheng Su, Zhiyuan Liu 0001, Peng Li 0030, Maosong Sun 0001, Jie Zhou 0016 |
NAACL-HLT | 10 |
| 2022 | On Transferability of Prompt Tuning for Natural Language ProcessingabstractYusheng Su, Xiaozhi Wang, Yujia Qin, Chi-Min Chan, Yankai Lin, Huadong Wang, Kaiyue Wen, Zhiyuan Liu, Peng Li, Juanzi Li, Lei Hou, Maosong Sun, Jie Zhou. Proceedings of the 2022 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies. 2022. Yusheng Su, Xiaozhi Wang, Yujia Qin, Chi-Min Chan, Yankai Lin 0001, Kaiyue Wen, Zhiyuan Liu 0001, Peng Li 0030, Juan-Zi Li, Lei Hou 0001, Maosong Sun 0001, Jie Zhou 0016 |
NAACL-HLT | 12 |
| 2022 | A Unified Evaluation of Textual Backdoor Learning: Frameworks and BenchmarksabstractTextual backdoor attacks are a kind of practical threat to NLP systems. By injecting a backdoor in the training phase, the adversary could control model predictions via predefined triggers. As various attack and defense models have been proposed, it is of great significance to perform rigorous evaluations. However, we highlight two issues in previous backdoor learning evaluations: (1) The differences between real-world scenarios (e.g. releasing poisoned datasets or models) are neglected, and we argue that each scenario has its own constraints and concerns, thus requires specific evaluation protocols; (2) The evaluation metrics only consider whether the attacks could flip the models' predictions on poisoned samples and retain performances on benign samples, but ignore that poisoned samples should also be stealthy and semantic-preserving. To address these issues, we categorize existing works into three practical scenarios in which attackers release datasets, pre-trained models, and fine-tuned models respectively, then discuss their unique evaluation methodologies. On metrics, to completely evaluate poisoned samples, we use grammar error increase and perplexity difference for stealthiness, along with text similarity for validity. After formalizing the frameworks, we develop an open-source toolkit OpenBackdoor to foster the implementations and evaluations of textual backdoor learning. With this toolkit, we perform extensive experiments to benchmark attack and defense models under the suggested paradigm. To facilitate the underexplored defenses against poisoned datasets, we further propose CUBE, a simple yet strong clustering-based defense baseline. We hope that our frameworks and benchmarks could serve as the cornerstones for future model development and evaluations. Ganqu Cui, Lifan Yuan, Bingxiang He, Yangyi Chen, Zhiyuan Liu 0001, Maosong Sun 0001 |
NeurIPS | 6 |
| 2022 | Sparse Structure Search for Delta TuningabstractAdapting large pre-trained models (PTMs) through fine-tuning imposes prohibitive computational and storage burdens. Recent studies of delta tuning (DT), i.e., parameter-efficient tuning, find that only optimizing a small portion of parameters conditioned on PTMs could yield on-par performance compared to conventional fine-tuning. Generally, DT methods exquisitely design delta modules (DT modules) which could be applied to arbitrary fine-grained positions inside PTMs. However, the effectiveness of these fine-grained positions largely relies on sophisticated manual designation, thereby usually producing sub-optimal results. In contrast to the manual designation, we explore constructing DT modules in an automatic manner. We automatically \textbf{S}earch for the \textbf{S}parse \textbf{S}tructure of \textbf{Delta} Tuning (S$^3$Delta). Based on a unified framework of various DT methods, S$^3$Delta conducts the differentiable DT structure search through bi-level optimization and proposes shifted global sigmoid method to explicitly control the number of trainable parameters. Extensive experiments show that S$^3$Delta surpasses manual and random structures with less trainable parameters. The searched structures preserve more than 99\% fine-tuning performance with 0.01\% trainable parameters. Moreover, the advantage of S$^3$Delta is amplified with extremely low trainable parameters budgets (0.0009\%$\sim$0.01\%). The searched structures are transferable and explainable, providing suggestions and guidance for the future design of DT methods. Our codes are publicly available at \url{https://github.com/thunlp/S3Delta}. Shengding Hu, Zhen Zhang 0008, Ning Ding 0002, Yadao Wang, Yasheng Wang, Zhiyuan Liu 0001, Maosong Sun 0001 |
NeurIPS | 7 |
| 2022 | Moderate-fitting as a Natural Backdoor Defender for Pre-trained Language ModelsabstractDespite the great success of pre-trained language models (PLMs) in a large set of natural language processing (NLP) tasks, there has been a growing concern about their security in real-world applications. Backdoor attack, which poisons a small number of training samples by inserting backdoor triggers, is a typical threat to security. Trained on the poisoned dataset, a victim model would perform normally on benign samples but predict the attacker-chosen label on samples containing pre-defined triggers. The vulnerability of PLMs under backdoor attacks has been proved with increasing evidence in the literature. In this paper, we present several simple yet effective training strategies that could effectively defend against such attacks. To the best of our knowledge, this is the first work to explore the possibility of backdoor-free adaptation for PLMs. Our motivation is based on the observation that, when trained on the poisoned dataset, the PLM's adaptation follows a strict order of two stages: (1) a moderate-fitting stage, where the model mainly learns the major features corresponding to the original task instead of subsidiary features of backdoor triggers, and (2) an overfitting stage, where both features are learned adequately. Therefore, if we could properly restrict the PLM's adaptation to the moderate-fitting stage, the model would neglect the backdoor triggers but still achieve satisfying performance on the original task. To this end, we design three methods to defend against backdoor attacks by reducing the model capacity, training epochs, and learning rate, respectively. Experimental results demonstrate the effectiveness of our methods in defending against several representative NLP backdoor attacks. We also perform visualization-based analysis to attain a deeper understanding of how the model learns different features, and explore the effect of the poisoning ratio. Finally, we explore whether our methods could defend against backdoor attacks for the pre-trained CV model. The codes are publicly available at https://github.com/thunlp/Moderate-fitting. Biru Zhu, Yujia Qin, Ganqu Cui, Yangyi Chen, Weilin Zhao, Yangdong Deng, Zhiyuan Liu 0001, Jingang Wang, Wei Wu 0014, Maosong Sun 0001, Ming Gu 0001 |
NeurIPS | 11 |
| 2022 | Data augmentation for low-resource languages NMT guided by constrained sampling
Mieradilijiang Maimaiti, Yang Liu 0005, Huan-Bo Luan, Maosong Sun 0001 |
Int. J. Intell. Syst. | 4 |
| 2022 | Dynamic Multi-Branch Layers for On-Device Neural Machine TranslationabstractWith the rapid development of artificial intelligence (AI), there is a trend in moving AI applications, such as neural machine translation (NMT), from cloud to mobile devices. Constrained by limited hardware resources and battery, the performance of on-device NMT systems is far from satisfactory. Inspired by conditional computation, we propose to improve the performance of on-device NMT systems with dynamic multi-branch layers. Specifically, we design a layer-wise dynamic multi-branch network with only one branch activated during training and inference. As not all branches are activated during training, we propose shared-private reparameterization to ensure sufficient training for each branch. At almost the same computational cost, our method achieves improvements of up to 1.7 BLEU points on the WMT14 English-German translation task and 1.8 BLEU points on the WMT20 Chinese-English translation task over the Transformer model, respectively. Compared with a strong baseline that also uses multiple branches, the proposed method is up to 1.5 times faster with the same number of parameters. Zhixing Tan, Zeyuan Yang 0002, Meng Zhang 0019, Qun Liu 0001, Maosong Sun 0001, Yang Liu 0005 |
IEEE ACM Trans. Audio Speech Lang. Process. | 5 |
| 2022 | COSINE: Compressive Network Embedding on Large-Scale Information NetworksabstractThere is recently a surge in approaches that learn low-dimensional embeddings of nodes in networks. However, for large-scale real-world networks, it’s inefficient for existing approaches to store amounts of parameters in memory and update them edge by edge. With the knowledge that nodes having similar neighborhoods will be close to each other in the embedding space, we propose COSINE (COmpresSIve Network Embedding) algorithm, which reduces the memory footprint and accelerates the training process by parameter sharing among similar nodes. COSINE applies graph partitioning algorithms to networks and builds parameter sharing dependency of nodes based on the results of partitioning. In this way, COSINE injects prior knowledge about high-order structural information into models, which makes network embedding more efficient and effective. COSINE can be applied to anyembedding lookupmethod and learn high-quality embeddings with limited memory and less training time. We conduct experiments on multi-label classification and link prediction, where baselines and our model have the same memory usage. Experimental results show that COSINE improves baselines by up to 23 percent on classification and 25 percent on link prediction. Moreover, the training time of all representation learning methods using COSINE decreases by 30 to 70 percent. Zhengyan Zhang, Cheng Yang 0002, Zhiyuan Liu 0001, Maosong Sun 0001, Zhichong Fang, Bo Zhang 0056, Leyu Lin |
IEEE Trans. Knowl. Data Eng. | 4 |
| 2022 | A Unified Understanding of Deep NLP Models for Text ClassificationabstractThe rapid development of deep natural language processing (NLP) models for text classification has led to an urgent need for a unified understanding of these models proposed individually. Existing methods cannot meet the need for understanding different models in one framework due to the lack of a unified measure for explaining both low-level (e.g., words) and high-level (e.g., phrases) features. We have developed a visual analysis tool, DeepNLPVis, to enable a unified understanding of NLP models for text classification. The key idea is a mutual information-based measure, which provides quantitative explanations on how each layer of a model maintains the information of input words in a sample. We model the intra- and inter-word information at each layer measuring the importance of a word to the final prediction as well as the relationships between words, such as the formation of phrases. A multi-level visualization, which consists of a corpus-level, a sample-level, and a word-level visualization, supports the analysis from the overall training set to individual samples. Two case studies on classification tasks and comparison between models demonstrate that DeepNLPVis can help users effectively identify potential problems caused by samples and model architectures and then make informed improvements. Zhen Li 0044, Xiting Wang, Weikai Yang, Jing Wu 0004, Zhengyan Zhang, Zhiyuan Liu 0001, Maosong Sun 0001, Hui Zhang 0013, Shixia Liu |
IEEE Trans. Vis. Comput. Graph. | 7 |
| 2021 | Aspect-Level Sentiment-Controllable Review Generation with Mutual Learning FrameworkabstractReview generation, aiming to automatically generate review text according to the given information, is proposed to assist in the unappealing review writing. However, most of existing methods only consider the overall sentiments of reviews and cannot achieve aspect-level sentiment control. Even though some previous studies attempt to generate aspect-level sentiment-controllable reviews, they usually require large-scale human annotations which are unavailable in the real world. To address this issue, we propose a mutual learning framework to take advantage of unlabeled data to assist the aspect-level sentiment-controllable review generation. The framework consists of a generator and a classifier which utilize confidence mechanism and reconstruction reward to enhance each other. Experimental results show our model can achieve aspect-sentiment control accuracy up to 88% without losing generation quality. Yankai Lin 0001, Fanchao Qi, Jinyi Hu, Peng Li 0030, Jie Zhou 0016, Maosong Sun 0001 |
AAAI | 7 |
| 2021 | Adversarial Language Games for Advanced Natural Language IntelligenceabstractWe study the problem of adversarial language games, in which multiple agents with conflicting goals compete with each other via natural language interactions. While adversarial language games are ubiquitous in human activities, little attention has been devoted to this field in natural language processing. In this work, we propose a challenging adversarial language game called Adversarial Taboo as an example, in which an attacker and a defender compete around a target word. The attacker is tasked with inducing the defender to utter the target word invisible to the defender, while the defender is tasked with detecting the target word before being induced by the attacker. In Adversarial Taboo, a successful attacker and defender need to hide or infer the intention, and induce or defend during conversations. This requires several advanced language abilities, such as adversarial pragmatic reasoning and goal-oriented language interactions in open domain, which will facilitate many downstream NLP tasks. To instantiate the game, we create a game environment and a competition platform. Comprehensive experiments on several baseline attack and defense strategies show promising and interesting results, based on which we discuss some directions for future research. Yuan Yao 0013, Haoxi Zhong, Zhengyan Zhang, Xu Han 0007, Xiaozhi Wang, Kai Zhang 0033, Chaojun Xiao, Guoyang Zeng, Zhiyuan Liu 0001, Maosong Sun 0001 |
AAAI | 10 |
| 2021 | Mask-Align: Self-Supervised Neural Word AlignmentabstractChi Chen, Maosong Sun, Yang Liu. 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. Chi Chen 0005, Maosong Sun 0001, Yang Liu 0005 |
ACL/IJCNLP (1) | 2 |
| 2021 | Transfer Learning for Sequence Generation: from Single-source to Multi-sourceabstractXuancheng Huang, Jingfang Xu, Maosong Sun, Yang Liu. 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. Xuancheng Huang, Jingfang Xu, Maosong Sun 0001, Yang Liu 0005 |
ACL/IJCNLP (1) | 3 |
| 2021 | Hidden Killer: Invisible Textual Backdoor Attacks with Syntactic TriggerabstractFanchao Qi, Mukai Li, Yangyi Chen, Zhengyan Zhang, Zhiyuan Liu, Yasheng Wang, Maosong Sun. 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. Fanchao Qi, Mukai Li, Yangyi Chen, Zhengyan Zhang, Zhiyuan Liu 0001, Yasheng Wang, Maosong Sun 0001 |
ACL/IJCNLP (1) | 7 |
| 2021 | Turn the Combination Lock: Learnable Textual Backdoor Attacks via Word SubstitutionabstractFanchao Qi, Yuan Yao, Sophia Xu, Zhiyuan Liu, Maosong Sun. 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. Fanchao Qi, Yuan Yao 0013, Sophia Xu, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL/IJCNLP (1) | 5 |
| 2021 | ERICA: Improving Entity and Relation Understanding for Pre-trained Language Models via Contrastive LearningabstractYujia Qin, Yankai Lin, Ryuichi Takanobu, Zhiyuan Liu, Peng Li, Heng Ji, Minlie Huang, Maosong Sun, Jie Zhou. 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. Yujia Qin, Yankai Lin 0001, Ryuichi Takanobu, Zhiyuan Liu 0001, Peng Li 0030, Heng Ji 0001, Minlie Huang, Maosong Sun 0001, Jie Zhou 0016 |
ACL/IJCNLP (1) | 8 |
| 2021 | MOOCCubeX: A Large Knowledge-centered Repository for Adaptive Learning in MOOCsabstractThe prosperity of massive open online courses provides fodder for plentiful research efforts on adaptive learning. However, current open-access educational datasets are still far from sufficient to meet the need for various topics of adaptive learning. Existing released datasets often cover only small-scale data, lack fine-grained knowledge concepts. They are even difficult to curate and supplement due to platform limitations. In this work, we construct MOOCCubeX, a large, knowledge-centered repository consisting of 4,216 courses, 230,263 videos, 358,265 exercises, 637,572 fine-grained concepts and over 296 million behavioral data of 3,330,294 students, for supporting the research topics on adaptive learning in MOOCs. Licensed by XuetangX, one of the largest MOOC websites in China, we obtain abundant and diverse course resources and student behavioral data and are permitted to make subsequent periodic updates. We propose a framework to accomplish data processing, weakly supervised fine-grained concept graph mining, and data curation to improve usability and richness. Based on the fine-grained concepts, we re-organize the data from the knowledge perspective and acquire more external learning resources from the web. Our repository is now available at https://github.com/THU-KEG/MOOCCubeX. Jifan Yu, Yuquan Wang, Qingyang Zhong, Gan Luo, Yiming Mao 0005, Wenzheng Feng, Wei Xu 0017, Shulin Cao, Kaisheng Zeng, Zijun Yao 0002, Lei Hou 0001, Yankai Lin 0001, Peng Li 0030, Jie Zhou 0016, Bin Xu 0001, Juan-Zi Li, Jie Tang 0001, Maosong Sun 0001 |
CIKM | 19 |
| 2021 | Segment, Mask, and Predict: Augmenting Chinese Word Segmentation with Self-SupervisionabstractMieradilijiang Maimaiti, Yang Liu, Yuanhang Zheng, Gang Chen, Kaiyu Huang, Ji Zhang, Huanbo Luan, Maosong Sun. Proceedings of the 2021 Conference on Empirical Methods in Natural Language Processing. 2021. Mieradilijiang Maimaiti, Yang Liu 0005, Yuanhang Zheng, Gang Chen 0039, Ji Zhang 0011, Huan-Bo Luan, Maosong Sun 0001 |
EMNLP (1) | 8 |
| 2021 | ONION: A Simple and Effective Defense Against Textual Backdoor AttacksabstractBackdoor attacks are a kind of emergent training-time threat to deep neural networks (DNNs).They can manipulate the output of DNNs and possess high insidiousness.In the field of natural language processing, some attack methods have been proposed and achieve very high attack success rates on multiple popular models.Nevertheless, there are few studies on defending against textual backdoor attacks.In this paper, we propose a simple and effective textual backdoor defense named ONION, which is based on outlier word detection and, to the best of our knowledge, is the first method that can handle all the textual backdoor attack situations.Experiments demonstrate the effectiveness of our model in defending BiLSTM and BERT against five different backdoor attacks.All the code and data of this paper can be obtained at https: //github.com/thunlp/ONION. Fanchao Qi, Yangyi Chen, Mukai Li, Yuan Yao 0013, Zhiyuan Liu 0001, Maosong Sun 0001 |
EMNLP (1) | 6 |
| 2021 | Mind the Style of Text! Adversarial and Backdoor Attacks Based on Text Style TransferabstractAdversarial attacks and backdoor attacks are two common security threats that hang over deep learning.Both of them harness taskirrelevant features of data in their implementation.Text style is a feature that is naturally irrelevant to most NLP tasks, and thus suitable for adversarial and backdoor attacks.In this paper, we make the first attempt to conduct adversarial and backdoor attacks based on text style transfer, which is aimed at altering the style of a sentence while preserving its meaning.We design an adversarial attack method and a backdoor attack method, and conduct extensive experiments to evaluate them.Experimental results show that popular NLP models are vulnerable to both adversarial and backdoor attacks based on text style transfer-the attack success rates can exceed 90% without much effort.It reflects the limited ability of NLP models to handle the feature of text style that has not been widely realized.In addition, the style transfer-based adversarial and backdoor attack methods show superiority to baselines in many aspects.All the code and data of this paper can be obtained at https:// github.com/thunlp/StyleAttack. Fanchao Qi, Yangyi Chen, Xurui Zhang, Mukai Li, Zhiyuan Liu 0001, Maosong Sun 0001 |
EMNLP (1) | 6 |
| 2021 | CodRED: A Cross-Document Relation Extraction Dataset for Acquiring Knowledge in the WildabstractExisting relation extraction (RE) methods typically focus on extracting relational facts between entity pairs within single sentences or documents.However, a large quantity of relational facts in knowledge bases can only be inferred across documents in practice.In this work, we present the problem of crossdocument RE, making an initial step towards knowledge acquisition in the wild.To facilitate the research, we construct the first human-annotated cross-document RE dataset CodRED.Compared to existing RE datasets, CodRED presents two key challenges: Given two entities, (1) it requires finding the relevant documents that can provide clues for identifying their relations; (2) it requires reasoning over multiple documents to extract the relational facts.We conduct comprehensive experiments to show that CodRED is challenging to existing RE methods including strong BERT-based models.We make CodRED and the code for our baselines publicly available at https://github.com/thunlp/CodRED. Yuan Yao 0013, Jiaju Du, Yankai Lin 0001, Peng Li 0030, Zhiyuan Liu 0001, Jie Zhou 0016, Maosong Sun 0001 |
EMNLP (1) | 7 |
| 2021 | Self-Supervised Quality Estimation for Machine TranslationabstractYuanhang Zheng, Zhixing Tan, Meng Zhang, Mieradilijiang Maimaiti, Huanbo Luan, Maosong Sun, Qun Liu, Yang Liu. Proceedings of the 2021 Conference on Empirical Methods in Natural Language Processing. 2021. Yuanhang Zheng, Zhixing Tan, Meng Zhang 0019, Mieradilijiang Maimaiti, Huan-Bo Luan, Maosong Sun 0001, Qun Liu 0001, Yang Liu 0005 |
EMNLP (1) | 6 |
| 2021 | Visual Distant Supervision for Scene Graph GenerationabstractScene graph generation aims to identify objects and their relations in images, providing structured image representations that can facilitate numerous applications in computer vision. However, scene graph models usually require supervised learning on large quantities of labeled data with intensive human annotation. In this work, we propose visual distant supervision, a novel paradigm of visual relation learning, which can train scene graph models without any human-labeled data. The intuition is that by aligning commonsense knowledge bases and images, we can automatically create large-scale labeled data to provide distant supervision for visual relation learning. To alleviate the noise in distantly labeled data, we further propose a framework that iteratively estimates the probabilistic relation labels and eliminates the noisy ones. Comprehensive experimental results show that our distantly supervised model outperforms strong weakly supervised and semi-supervised baselines. By further incorporating human-labeled data in a semi-supervised fashion, our model outperforms state-of-the-art fully supervised models by a large margin (e.g., 8.3 micro- and 7.8 macro-recall@50 improvements for predicate classification in Visual Genome evaluation). We make the data and code for this paper publicly available at https://github.com/thunlp/VisualDS. Yuan Yao 0011, Xu Han 0007, Mengdi Li 0006, Cornelius Weber, Zhiyuan Liu 0001, Stefan Wermter, Maosong Sun 0001 |
ICCV | 8 |
| 2021 | Neural Quality Estimation with Multiple Hypotheses for Grammatical Error CorrectionabstractZhenghao Liu, Xiaoyuan Yi, Maosong Sun, Liner Yang, Tat-Seng Chua. Proceedings of the 2021 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies. 2021. Zhenghao Liu 0001, Xiaoyuan Yi, Maosong Sun 0001, Liner Yang, Tat-Seng Chua |
NAACL-HLT | 3 |
| 2021 | TR-BERT: Dynamic Token Reduction for Accelerating BERT InferenceabstractExisting pre-trained language models (PLMs) are often computationally expensive in inference, making them impractical in various resource-limited real-world applications.To address this issue, we propose a dynamic token reduction approach to accelerate PLMs' inference, named TR-BERT, which could flexibly adapt the layer number of each token in inference to avoid redundant calculation.Specially, TR-BERT formulates the token reduction process as a multi-step token selection problem and automatically learns the selection strategy via reinforcement learning.The experimental results on several downstream NLP tasks show that TR-BERT is able to speed up BERT by 2-5 times to satisfy various performance demands.Moreover, TR-BERT can also achieve better performance with less computation in a suite of long-text tasks since its token-level layer number adaption greatly accelerates the self-attention operation in PLMs.The source code and experiment details of this paper can be obtained from https://github.com/ thunlp/TR-BERT. Deming Ye, Yankai Lin 0001, Yufei Huang 0008, Maosong Sun 0001 |
NAACL-HLT | 4 |
| 2021 | Open Hierarchical Relation ExtractionabstractKai Zhang, Yuan Yao, Ruobing Xie, Xu Han, Zhiyuan Liu, Fen Lin, Leyu Lin, Maosong Sun. Proceedings of the 2021 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies. 2021. Kai Zhang 0033, Yuan Yao 0013, Ruobing Xie, Xu Han 0007, Zhiyuan Liu 0001, Fen Lin 0002, Leyu Lin, Maosong Sun 0001 |
NAACL-HLT | 8 |
| 2021 | OpenMatch: An Open Source Library for Neu-IR ResearchabstractOpenMatch is a Python-based library that serves for Neural Information Retrieval (Neu-IR) research. It provides self-contained neural and traditional IR modules, making it easy to build customized and higher-capacity IR systems. In order to develop the advantages of Neu-IR models for users, OpenMatch provides implementations of recent neural IR models, complicated experiment instructions, and advanced few-shot training methods. OpenMatch reproduces corresponding ranking results of previous work on widely-used IR benchmarks, liberating users from surplus labor in baseline reimplementation. Our OpenMatch-based solutions conduct top-ranked empirical results on various ranking tasks, such as ad hoc retrieval and conversational retrieval, illustrating the convenience of OpenMatch to facilitate building an effective IR system. The library, experimental methodologies and results of OpenMatch are all publicly available at https://github.com/thunlp/OpenMatch. Zhenghao Liu 0001, Kaitao Zhang, Chenyan Xiong, Zhiyuan Liu 0001, Maosong Sun 0001 |
SIGIR | 5 |
| 2021 | Sememe knowledge computation: a review of recent advances in application and expansion of sememe knowledge bases
Fanchao Qi, Ruobing Xie, Yuan Zang, Zhiyuan Liu 0001, Maosong Sun 0001 |
Frontiers Comput. Sci. | 5 |
| 2021 | Improving Data Augmentation for Low-Resource NMT Guided by POS-Tagging and Paraphrase EmbeddingabstractData augmentation is an approach for several text generation tasks. Generally, in the machine translation paradigm, mainly in low-resource language scenarios, many data augmentation methods have been proposed. The most used approaches for generating pseudo data mainly lay in word omission, random sampling, or replacing some words in the text. However, previous methods barely guarantee the quality of augmented data. In this work, we try to build the data by using paraphrase embedding and POS-Tagging. Namely, we generate the fake monolingual corpus by replacing the main four POS-Tagging labels, such as noun, adjective, adverb, and verb, based on both the paraphrase table and their similarity. We select the bigger corpus size of the paraphrase table with word level and obtain the word embedding of each word in the table, then calculate the cosine similarity between these words and tagged words in the original sequence. In addition, we exploit the ranking algorithm to choose highly similar words to reduce semantic errors and leverage the POS-Tagging replacement to mitigate syntactic error to some extent. Experimental results show that our augmentation method consistently outperforms all previous SOTA methods on the low-resource language pairs in seven language pairs from four corpora by 1.16 to 2.39 BLEU points. Mieradilijiang Maimaiti, Yang Liu 0005, Huan-Bo Luan, Zegao Pan, Maosong Sun 0001 |
ACM Trans. Asian Low Resour. Lang. Inf. Process. | 5 |
| 2021 | Learning to Generate Explainable Plots for Neural Story GenerationabstractStory generation is an important natural language processing task that aims to generate coherent stories automatically. While the use of neural networks has proven effective in improving story generation, how to learn to generate an explainable high-level plot still remains a major challenge. In this article, we propose a latent variable model for neural story generation. The model treats an outline, which is a natural language sentence explainable to humans, as a latent variable to represent a high-level plot that bridges the input and output. We adopt an external summarization model to guide the latent variable model to learn how to generate outlines from training data. Experiments show that our approach achieves significant improvements over state-of-the-art methods in both automatic and human evaluations. Gang Chen 0039, Yang Liu 0005, Huan-Bo Luan, Meng Zhang 0019, Qun Liu 0001, Maosong Sun 0001 |
IEEE ACM Trans. Audio Speech Lang. Process. | 6 |
| 2021 | CSS-LM: A Contrastive Framework for Semi-Supervised Fine-Tuning of Pre-Trained Language ModelsabstractFine-tuning pre-trained language models (PLMs) has demonstrated its effectiveness on various downstream NLP tasks recently. However, in many scenarios with limited supervised data, the conventional fine-tuning strategies cannot sufficiently capture the important semantic features for downstream tasks. To address this issue, we introduce a novel framework (named ‘`CSS-LM’') to improve the fine-tuning phase of PLMs via contrastive semi-supervised learning. Specifically, given a specific task, we retrieve positive and negative instances from large-scale unlabeled corpora according to their domain-level and class-level semantic relatedness to the task. We then perform contrastive semi-supervised learning on both the retrieved unlabeled instances and original labeled instances to help PLMs capture crucial task-related semantic features. The experimental results show that CSS-LM achieves better results than the conventional fine-tuning strategy on a series of downstream tasks with few-shot settings by up to 7.8%, and outperforms the latest supervised contrastive fine-tuning strategy by up to 7.1%. Our datasets and source code will be available to provide more details. Yusheng Su, Xu Han 0007, Yankai Lin 0001, Zhengyan Zhang, Zhiyuan Liu 0001, Peng Li 0030, Jie Zhou 0016, Maosong Sun 0001 |
IEEE ACM Trans. Audio Speech Lang. Process. | 8 |
| 2021 | Neural Machine Translation With Explicit Phrase AlignmentabstractWhile neural machine translation has achieved state-of-the-art translation performance, it is unable to capture the alignment between the input and output during the translation process. The lack of alignment in neural machine translation models leads to three problems: it is hard to (1) interpret the translation process, (2) impose lexical constraints, and (3) impose structural constraints. These problems not only increase the difficulty of designing new architectures for neural machine translation, but also limit its applications in practice. To alleviate these problems, we propose to introduce explicit phrase alignment into the translation process of arbitrary neural machine translation models. The key idea is to build a search space similar to that of phrase-based statistical machine translation for neural machine translation where phrase alignment is readily available. We design a new decoding algorithm that can easily impose lexical and structural constraints. Experiments show that our approach makes the translation process of neural machine translation more interpretable without sacrificing translation quality. In addition, our approach achieves significant improvements in lexically and structurally constrained translation tasks. Huan-Bo Luan, Maosong Sun 0001, Feifei Zhai, Jingfang Xu, Yang Liu 0005 |
IEEE ACM Trans. Audio Speech Lang. Process. | 3 |
| 2021 | Country Image in COVID-19 Pandemic: A Case Study of ChinaabstractCountry image has a profound influence on international relations and economic development. In the worldwide outbreak of COVID-19, countries and their people display different reactions, resulting in diverse perceived images among foreign public. Therefore, in this article, we take China as a specific and typical case and investigate its image with aspect-based sentiment analysis on a large-scale Twitter dataset. To our knowledge, this is the first study to explore country image in such a fine-grained way. To perform the analysis, we first build a manually-labeled Twitter dataset with aspect-level sentiment annotations. Afterward, we conduct the aspect-based sentiment analysis with BERT to explore the image of China. We discover an overall sentiment change from non-negative to negative in the general public, and explain it with the increasing mentions of negative ideology-related aspects and decreasing mentions of non-negative fact-based aspects. Further investigations into different groups of Twitter users, including U.S. Congress members, English media, and social bots, reveal different patterns in their attitudes toward China. This article provides a deeper understanding of the changing image of China in COVID-19 pandemic. Our research also demonstrates how aspect-based sentiment analysis can be applied in social science researches to deliver valuable insights. Fanchao Qi, Yining Ye, Zhiyuan Liu 0001, Maosong Sun 0001, Jianbin Jin |
IEEE Trans. Big Data | 6 |
| 2021 | CED: Credible Early Detection of Social Media RumorsabstractRumors spread dramatically fast through online social media services, and people are exploring methods to detect rumors automatically. Existing methods typically learn semantic representations of all reposts to a rumor candidate for prediction. However, it is crucial to efficiently detect rumors as early as possible before they cause severe social disruption, which has not been well addressed by previous works. In this paper, we present a novel early rumor detection model, Credible Early Detection (CED). By regarding all reposts to a rumor candidate as a sequence, the proposed model will seek an early point-in-time for making a credible prediction. We conduct experiments on three real-world datasets, and the results demonstrate that our proposed model can remarkably reduce the time span for prediction by more than 85 percent, with better accuracy performance than all state-of-the-art baselines. Changhe Song, Cheng Yang 0002, Cunchao Tu, Zhiyuan Liu 0001, Maosong Sun 0001 |
IEEE Trans. Knowl. Data Eng. | 6 |
| 2021 | Neural Diffusion Model for Microscopic Cascade StudyabstractThe study of information diffusion or cascade has attracted much attention over the last decade. Most related works target on studying cascade-level macroscopic properties such as the final size of a cascade. Existing microscopic cascade models which focus on user-level modeling either make strong assumptions on how a user gets infected by a cascade or limit themselves to a specific scenario where “who infected whom” information is explicitly labeled. The strong assumptions oversimplify the complex diffusion mechanism and prevent these models from better fitting real-world cascade data. Also, the methods which focus on specific scenarios cannot be generalized to a general setting where the diffusion graph is unobserved. To overcome the drawbacks of previous works, we propose a Neural Diffusion Model (NDM) for general microscopic cascade study. NDM makes relaxed assumptions and employs deep learning techniques including attention mechanism and convolutional network for cascade modeling. Both advantages enable our model to go beyond the limitations of previous methods, better fit the diffusion data and generalize to unseen cascades. Experimental results on diffusion identification task over four realistic cascade datasets show that our model can achieve a relative improvement up to 26 percent against the best performing baseline in terms of F1 score. Cheng Yang 0002, Maosong Sun 0001, Shiyi Han, Zhiyuan Liu 0001, Huan-Bo Luan |
IEEE Trans. Knowl. Data Eng. | 2 |
| 2020 | Neural Snowball for Few-Shot Relation LearningabstractKnowledge graphs typically undergo open-ended growth of new relations. This cannot be well handled by relation extraction that focuses on pre-defined relations with sufficient training data. To address new relations with few-shot instances, we propose a novel bootstrapping approach, Neural Snowball, to learn new relations by transferring semantic knowledge about existing relations. More specifically, we use Relational Siamese Networks (RSN) to learn the metric of relational similarities between instances based on existing relations and their labeled data. Afterwards, given a new relation and its few-shot instances, we use RSN to accumulate reliable instances from unlabeled corpora; these instances are used to train a relation classifier, which can further identify new facts of the new relation. The process is conducted iteratively like a snowball. Experiments show that our model can gather high-quality instances for better few-shot relation learning and achieves significant improvement compared to baselines. Codes and datasets are released on https://github.com/thunlp/Neural-Snowball. Tianyu Gao 0001, Xu Han 0007, Ruobing Xie, Zhiyuan Liu 0001, Fen Lin 0002, Leyu Lin, Maosong Sun 0001 |
AAAI | 7 |
| 2020 | Towards Building a Multilingual Sememe Knowledge Base: Predicting Sememes for BabelNet SynsetsabstractA sememe is defined as the minimum semantic unit of human languages. Sememe knowledge bases (KBs), which contain words annotated with sememes, have been successfully applied to many NLP tasks. However, existing sememe KBs are built on only a few languages, which hinders their widespread utilization. To address the issue, we propose to build a unified sememe KB for multiple languages based on BabelNet, a multilingual encyclopedic dictionary. We first build a dataset serving as the seed of the multilingual sememe KB. It manually annotates sememes for over 15 thousand synsets (the entries of BabelNet). Then, we present a novel task of automatic sememe prediction for synsets, aiming to expand the seed dataset into a usable KB. We also propose two simple and effective models, which exploit different information of synsets. Finally, we conduct quantitative and qualitative analyses to explore important factors and difficulties in the task. All the source code and data of this work can be obtained on https://github.com/thunlp/BabelNet-Sememe-Prediction. Fanchao Qi, Liang Chang 0007, Maosong Sun 0001, Sicong Ouyang, Zhiyuan Liu 0001 |
AAAI | 3 |
| 2020 | MixPoet: Diverse Poetry Generation via Learning Controllable Mixed Latent SpaceabstractAs an essential step towards computer creativity, automatic poetry generation has gained increasing attention these years. Though recent neural models make prominent progress in some criteria of poetry quality, generated poems still suffer from the problem of poor diversity. Related literature researches show that different factors, such as life experience, historical background, etc., would influence composition styles of poets, which considerably contributes to the high diversity of human-authored poetry. Inspired by this, we propose MixPoet, a novel model that absorbs multiple factors to create various styles and promote diversity. Based on a semi-supervised variational autoencoder, our model disentangles the latent space into some subspaces, with each conditioned on one influence factor by adversarial training. In this way, the model learns a controllable latent variable to capture and mix generalized factor-related properties. Different factor mixtures lead to diverse styles and hence further differentiate generated poems from each other. Experiment results on Chinese poetry demonstrate that MixPoet improves both diversity and quality against three state-of-the-art models. Xiaoyuan Yi, Cheng Yang 0002, Wenhao Li 0003, Maosong Sun 0001 |
AAAI | 5 |
| 2020 | Multi-Channel Reverse Dictionary ModelabstractA reverse dictionary takes the description of a target word as input and outputs the target word together with other words that match the description. Existing reverse dictionary methods cannot deal with highly variable input queries and low-frequency target words successfully. Inspired by the description-to-word inference process of humans, we propose the multi-channel reverse dictionary model, which can mitigate the two problems simultaneously. Our model comprises a sentence encoder and multiple predictors. The predictors are expected to identify different characteristics of the target word from the input query. We evaluate our model on English and Chinese datasets including both dictionary definitions and human-written descriptions. Experimental results show that our model achieves the state-of-the-art performance, and even outperforms the most popular commercial reverse dictionary system on the human-written description dataset. We also conduct quantitative analyses and a case study to demonstrate the effectiveness and robustness of our model. All the code and data of this work can be obtained on https://github.com/thunlp/MultiRD. Fanchao Qi, Zhiyuan Liu 0001, Yasheng Wang, Qun Liu 0001, Maosong Sun 0001 |
AAAI | 6 |
| 2020 | Iteratively Questioning and Answering for Interpretable Legal Judgment PredictionabstractLegal Judgment Prediction (LJP) aims to predict judgment results according to the facts of cases. In recent years, LJP has drawn increasing attention rapidly from both academia and the legal industry, as it can provide references for legal practitioners and is expected to promote judicial justice. However, the research to date usually suffers from the lack of interpretability, which may lead to ethical issues like inconsistent judgments or gender bias. In this paper, we present QAjudge, a model based on reinforcement learning to visualize the prediction process and give interpretable judgments. QAjudge follows two essential principles in legal systems across the world: Presumption of Innocence and Elemental Trial. During inference, a Question Net will select questions from the given set and an Answer Net will answer the question according to the fact description. Finally, a Predict Net will produce judgment results based on the answers. Reward functions are designed to minimize the number of questions asked. We conduct extensive experiments on several real-world datasets. Experimental results show that QAjudge can provide interpretable judgments while maintaining comparable performance with other state-of-the-art LJP models. The codes can be found from https://github.com/thunlp/QAjudge. Haoxi Zhong, Cunchao Tu, Tianyang Zhang 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
AAAI | 6 |
| 2020 | JEC-QA: A Legal-Domain Question Answering DatasetabstractWe present JEC-QA, the largest question answering dataset in the legal domain, collected from the National Judicial Examination of China. The examination is a comprehensive evaluation of professional skills for legal practitioners. College students are required to pass the examination to be certified as a lawyer or a judge. The dataset is challenging for existing question answering methods, because both retrieving relevant materials and answering questions require the ability of logic reasoning. Due to the high demand of multiple reasoning abilities to answer legal questions, the state-of-the-art models can only achieve about 28% accuracy on JEC-QA, while skilled humans and unskilled humans can reach 81% and 64% accuracy respectively, which indicates a huge gap between humans and machines on this task. We will release JEC-QA and our baselines to help improve the reasoning ability of machine comprehension models. You can access the dataset from http://jecqa.thunlp.org/. Haoxi Zhong, Chaojun Xiao, Cunchao Tu, Tianyang Zhang 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
AAAI | 6 |
| 2020 | Continual Relation Learning via Episodic Memory Activation and ReconsolidationabstractContinual relation learning aims to continually train a model on new data to learn incessantly emerging novel relations while avoiding catastrophically forgetting old relations.Some pioneering work has proved that storing a handful of historical relation examples in episodic memory and replaying them in subsequent training is an effective solution for such a challenging problem.However, these memorybased methods usually suffer from overfitting the few memorized examples of old relations, which may gradually cause inevitable confusion among existing relations.Inspired by the mechanism in human long-term memory formation, we introduce episodic memory activation and reconsolidation (EMAR) to continual relation learning.Every time neural models are activated to learn both new and memorized data, EMAR utilizes relation prototypes for memory reconsolidation exercise to keep a stable understanding of old relations.The experimental results show that EMAR could get rid of catastrophically forgetting old relations and outperform the state-of-the-art continual learning models.The code and datasets are released on https://github.com/thunlp/ ContinualRE. Xu Han 0007, Tianyu Gao 0001, Yankai Lin 0001, Zhiyuan Liu 0001, Peng Li 0030, Maosong Sun 0001, Jie Zhou 0016 |
ACL | 7 |
| 2020 | Fine-grained Fact Verification with Kernel Graph Attention NetworkabstractFact Verification requires fine-grained natural language inference capability that finds subtle clues to identify the syntactical and semantically correct but not well-supported claims.This paper presents Kernel Graph Attention Network (KGAT), which conducts more finegrained fact verification with kernel-based attentions.Given a claim and a set of potential evidence sentences that form an evidence graph, KGAT introduces node kernels, which better measure the importance of the evidence node, and edge kernels, which conduct fine-grained evidence propagation in the graph, into Graph Attention Networks for more accurate fact verification.KGAT achieves a 70.38%FEVER score and significantly outperforms existing fact verification models on FEVER, a large-scale benchmark for fact verification.Our analyses illustrate that, compared to dot-product attentions, the kernelbased attention concentrates more on relevant evidence sentences and meaningful clues in the evidence graph, which is the main source of KGAT's effectiveness.All source codes of this work are available at https://github. com/thunlp/KernelGAT. Zhenghao Liu 0001, Chenyan Xiong, Maosong Sun 0001, Zhiyuan Liu 0001 |
ACL | 3 |
| 2020 | Word-level Textual Adversarial Attacking as Combinatorial OptimizationabstractAdversarial attacks are carried out to reveal the vulnerability of deep neural networks.Textual adversarial attacking is challenging because text is discrete and a small perturbation can bring significant change to the original input.Word-level attacking, which can be regarded as a combinatorial optimization problem, is a well-studied class of textual attack methods.However, existing word-level attack models are far from perfect, largely because unsuitable search space reduction methods and inefficient optimization algorithms are employed.In this paper, we propose a novel attack model, which incorporates the sememebased word substitution method and particle swarm optimization-based search algorithm to solve the two problems separately.We conduct exhaustive experiments to evaluate our attack model by attacking BiLSTM and BERT on three benchmark datasets.Experimental results demonstrate that our model consistently achieves much higher attack success rates and crafts more high-quality adversarial examples as compared to baseline methods.Also, further experiments show our model has higher transferability and can bring more robustness enhancement to victim models by adversarial training. Yuan Zang, Fanchao Qi, Chenghao Yang 0001, Zhiyuan Liu 0001, Meng Zhang 0019, Qun Liu 0001, Maosong Sun 0001 |
ACL | 7 |
| 2020 | How Does NLP Benefit Legal System: A Summary of Legal Artificial IntelligenceabstractLegal Artificial Intelligence (LegalAI) focuses on applying the technology of artificial intelligence, especially natural language processing, to benefit tasks in the legal domain.In recent years, LegalAI has drawn increasing attention rapidly from both AI researchers and legal professionals, as LegalAI is beneficial to the legal system for liberating legal professionals from a maze of paperwork.Legal professionals often think about how to solve tasks from rulebased and symbol-based methods, while NLP researchers concentrate more on data-driven and embedding methods.In this paper, we describe the history, the current state, and the future directions of research in LegalAI.We illustrate the tasks from the perspectives of legal professionals and NLP researchers and show several representative applications in LegalAI.We conduct experiments and provide an indepth analysis of the advantages and disadvantages of existing works to explore possible future directions.You can find the implementation of our work from https://github. com/thunlp/CLAIM. Haoxi Zhong, Chaojun Xiao, Cunchao Tu, Tianyang Zhang 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL | 6 |
| 2020 | Meta-Information Guided Meta-Learning for Few-Shot Relation ClassificationabstractFew-shot classification requires classifiers to adapt to new classes with only a few training instances.State-of-the-art meta-learning approaches such as MAML learn how to initialize and fast adapt parameters from limited instances, which have shown promising results in few-shot classification.However, existing meta-learning models solely rely on implicit instance-based statistics, and thus suffer from instance unreliability and weak interpretability.To solve this problem, we propose a novel meta-information guided meta-learning (MIML) framework, where semantic concepts of classes provide strong guidance for meta-learning in both initialization and adaptation.In effect, our model can establish connections between instance-based information and semantic-based information, which enables more effective initialization and faster adaptation.Comprehensive experimental results on few-shot relation classification demonstrate the effectiveness of the proposed framework.Notably, MIML achieves comparable or superior performance to humans with only one shot on FewRel evaluation.The source code and experiment details of this paper can be obtained from https://github.com/thunlp/MIML. Bowen Dong 0005, Yuan Yao 0013, Ruobing Xie, Tianyu Gao 0001, Xu Han 0007, Zhiyuan Liu 0001, Fen Lin 0002, Leyu Lin, Maosong Sun 0001 |
COLING | 9 |
| 2020 | Try to Substitute: An Unsupervised Chinese Word Sense Disambiguation Method Based on HowNetabstractWord sense disambiguation (WSD) is a fundamental natural language processing task.Unsupervised knowledge-based WSD only relies on a lexical knowledge base as the sense inventory and has wider practical use than supervised WSD that requires a mass of sense-annotated data.HowNet is the most widely used lexical knowledge base in Chinese WSD.Because of its uniqueness, however, most of existing unsupervised WSD methods cannot work for HowNetbased WSD, and the tailor-made methods have not obtained satisfying results.In this paper, we propose a new unsupervised method for HowNet-based Chinese WSD, which exploits the masked language model task of pre-trained language models.In experiments, considering existing evaluation dataset is small and out-of-date, we build a new and larger HowNet-based WSD dataset.Experimental results demonstrate that our model achieves significantly better performance than all the baseline methods.All the code and data of this paper are available at https://github.com/thunlp/SememeWSD. Bairu Hou, Fanchao Qi, Yuan Zang, Xurui Zhang, Zhiyuan Liu 0001, Maosong Sun 0001 |
COLING | 6 |
| 2020 | Train No Evil: Selective Masking for Task-Guided Pre-TrainingabstractRecently, pre-trained language models mostly follow the pre-train-then-fine-tuning paradigm and have achieved great performance on various downstream tasks.However, since the pretraining stage is typically task-agnostic and the fine-tuning stage usually suffers from insufficient supervised data, the models cannot always well capture the domain-specific and task-specific patterns.In this paper, we propose a three-stage framework by adding a task-guided pre-training stage with selective masking between general pre-training and finetuning.In this stage, the model is trained by masked language modeling on in-domain unsupervised data to learn domain-specific patterns and we propose a novel selective masking strategy to learn task-specific patterns.Specifically, we design a method to measure the importance of each token in sequences and selectively mask the important tokens.Experimental results on two sentiment analysis tasks show that our method can achieve comparable or even better performance with less than 50% of computation cost, which indicates our method is both effective and efficient.The source code of this paper can be obtained from https://github. com/thunlp/SelectiveMasking. Yuxian Gu, Zhengyan Zhang, Xiaozhi Wang, Zhiyuan Liu 0001, Maosong Sun 0001 |
EMNLP (1) | 5 |
| 2020 | Learning from Context or Names? An Empirical Study on Neural Relation ExtractionabstractNeural models have achieved remarkable success on relation extraction (RE) benchmarks.However, there is no clear understanding which type of information affects existing RE models to make decisions and how to further improve the performance of these models.To this end, we empirically study the effect of two main information sources in text: textual context and entity mentions (names).We find that (i) while context is the main source to support the predictions, RE models also heavily rely on the information from entity mentions, most of which is type information, and (ii) existing datasets may leak shallow heuristics via entity mentions and thus contribute to the high performance on RE benchmarks.Based on the analyses, we propose an entity-masked contrastive pre-training framework for RE to gain a deeper understanding on both textual context and type information while avoiding rote memorization of entities or use of superficial cues in mentions.We carry out extensive experiments to support our views, and show that our framework can improve the effectiveness and robustness of neural models in different RE scenarios.All the code and datasets are released at https://github.com/thunlp/ Hao Peng 0015, Tianyu Gao 0001, Xu Han 0007, Yankai Lin 0001, Peng Li 0030, Zhiyuan Liu 0001, Maosong Sun 0001, Jie Zhou 0016 |
EMNLP (1) | 7 |
| 2020 | Denoising Relation Extraction from Document-level Distant SupervisionabstractDistant supervision (DS) has been widely used to generate auto-labeled data for sentencelevel relation extraction (RE), which improves RE performance.However, the existing success of DS cannot be directly transferred to the more challenging document-level relation extraction (DocRE), since the inherent noise in DS may be even multiplied in document level and significantly harm the performance of RE.To address this challenge, we propose a novel pre-trained model for DocRE, which denoises the document-level DS data via multiple pre-training tasks.Experimental results on the large-scale DocRE benchmark show that our model can capture useful information from noisy DS data and achieve promising results.The source code of this paper can be found in https://github.com/thunlp/DSDocRE. Chaojun Xiao, Yuan Yao 0013, Ruobing Xie, Xu Han 0007, Zhiyuan Liu 0001, Maosong Sun 0001, Fen Lin 0002, Leyu Lin |
EMNLP (1) | 6 |
| 2020 | Coreferential Reasoning Learning for Language RepresentationabstractLanguage representation models such as BERT could effectively capture contextual semantic information from plain text, and have been proved to achieve promising results in lots of downstream NLP tasks with appropriate fine-tuning.However, most existing language representation models cannot explicitly handle coreference, which is essential to the coherent understanding of the whole discourse.To address this issue, we present CorefBERT, a novel language representation model that can capture the coreferential relations in context.The experimental results show that, compared with existing baseline models, CorefBERT can achieve significant improvements consistently on various downstream NLP tasks that require coreferential reasoning, while maintaining comparable performance to previous models on other common NLP tasks.The source code and experiment details of this paper can be obtained from https://github. com/thunlp/CorefBERT. Deming Ye, Yankai Lin 0001, Jiaju Du, Zhenghao Liu 0001, Peng Li 0030, Maosong Sun 0001, Zhiyuan Liu 0001 |
EMNLP (1) | 6 |
| 2020 | Modeling Voting for System Combination in Machine TranslationabstractSystem combination is an important technique for combining the hypotheses of different machine translation systems to improve translation performance. Although early statistical approaches to system combination have been proven effective in analyzing the consensus between hypotheses, they suffer from the error propagation problem due to the use of pipelines. While this problem has been alleviated by end-to-end training of multi-source sequence-to-sequence models recently, these neural models do not explicitly analyze the relations between hypotheses and fail to capture their agreement because the attention to a word in a hypothesis is calculated independently, ignoring the fact that the word might occur in multiple hypotheses. In this work, we propose an approach to modeling voting for system combination in machine translation. The basic idea is to enable words in hypotheses from different systems to vote on words that are representative and should get involved in the generation process. This can be done by quantifying the influence of each voter and its preference for each candidate. Our approach combines the advantages of statistical and neural methods since it can not only analyze the relations between hypotheses but also allow for end-to-end training. Experiments show that our approach is capable of better taking advantage of the consensus between hypotheses and achieves significant improvements over state-of-the-art baselines on Chinese-English and English-German machine translation tasks. Xuancheng Huang, Zhixing Tan, Derek F. Wong, Huan-Bo Luan, Jingfang Xu, Maosong Sun 0001, Yang Liu 0005 |
IJCAI | 7 |
| 2020 | Text Style Transfer via Learning Style Instance Supported Latent SpaceabstractText style transfer pursues altering the style of a sentence while remaining its main content unchanged. Due to the lack of parallel corpora, most recent work focuses on unsupervised methods and has achieved noticeable progress. Nonetheless, the intractability of completely disentangling content from style for text leads to a contradiction of content preservation and style transfer accuracy. To address this problem, we propose a style instance supported method, StyIns. Instead of representing styles with embeddings or latent variables learned from single sentences, our model leverages the generative flow technique to extract underlying stylistic properties from multiple instances of each style, which form a more discriminative and expressive latent style space. By combining such a space with the attention-based structure, our model can better maintain the content and simultaneously achieve high transfer accuracy. Furthermore, the proposed method can be flexibly extended to semi-supervised learning so as to utilize available limited paired data. Experiments on three transfer tasks, sentiment modification, formality rephrasing, and poeticness generation, show that StyIns obtains a better balance between content and style, outperforming several recent baselines. Xiaoyuan Yi, Zhenghao Liu 0001, Wenhao Li 0003, Maosong Sun 0001 |
IJCAI | 4 |
| 2020 | Generating Major Types of Chinese Classical Poetry in a Uniformed FrameworkabstractPoetry generation is an interesting research topic in the field of text generation. As one of the most valuable literary and cultural heritages of China, Chinese classical poetry is very familiar and loved by Chinese people from generation to generation. It has many particular characteristics in its language structure, ranging from form, sound to meaning, thus is regarded as an ideal testing task for text generation. In this paper, we propose a GPT-2 based uniformed framework for generating major types of Chinese classical poems. We define a unified format for formulating all types of training samples by integrating detailed form information, then present a simple form- stressed weighting method in GPT-2 to strengthen the control to the form of the generated poems, with special emphasis on those forms with longer body length. Preliminary experimental results show this enhanced model can generate Chinese classical poems of major types with high quality in both form and content, validating the effectiveness of the proposed strategy. The model has been incorporated into Jiuge, the most influential Chinese classical poetry generation system developed by Tsinghua University. Jinyi Hu, Maosong Sun 0001 |
LREC | 2 |
| 2020 | PiRhDy: Learning Pitch-, Rhythm-, and Dynamics-aware Embeddings for Symbolic MusicabstractDefinitive embeddings remain a fundamental challenge of computational musicology for symbolic music in deep learning today. Analogous to natural language, music can be modeled as a sequence of tokens. This motivates the majority of existing solutions to explore the utilization of word embedding models to build music embeddings. However, music differs from natural languages in two key aspects: (1) musical token is multi-faceted -- it comprises of pitch, rhythm and dynamics information; and (2) musical context is two-dimensional -- each musical token is dependent on both melodic and harmonic contexts. In this work, we provide a comprehensive solution by proposing a novel framework named PiRhDy that integrates pitch, rhythm, and dynamics information seamlessly. PiRhDy adopts a hierarchical strategy which can be decomposed into two steps: (1) token (i.e., note event) modeling, which separately represents pitch, rhythm, and dynamics and integrates them into a single token embedding; and (2) context modeling, which utilizes melodic and harmonic knowledge to train the token embedding. A thorough study was made on each component and sub-strategy of PiRhDy.We further validate our embeddings in three downstream tasks -- melody completion, accompaniment suggestion, and genre classification. Results indicate a significant advancement of the neural approach towards symbolic music as well as PiRhDy's potential as a pretrained tool for a broad range of symbolic music applications. Hongru Liang, Wenqiang Lei, Paul Y. Chan, Zhenglu Yang, Maosong Sun 0001, Tat-Seng Chua |
ACM Multimedia | 5 |
| 2020 | Towards Interpretable Natural Language Understanding with Explanations as Latent VariablesabstractRecently generating natural language explanations has shown very promising results in not only offering interpretable explanations but also providing additional information and supervision for prediction. However, existing approaches usually require a large set of human annotated explanations for training while collecting a large set of explanations is not only time consuming but also expensive. In this paper, we develop a general framework for interpretable natural language understanding that requires only a small set of human annotated explanations for training. Our framework treats natural language explanations as latent variables that model the underlying reasoning process of a neural model. We develop a variational EM framework for optimization where an explanation generation module and an explanation-augmented prediction module are alternatively optimized and mutually enhance each other. Moreover, we further propose an explanation-based self-training method under this framework for semi-supervised learning. It alternates between assigning pseudo-labels to unlabeled data and generating new explanations to iteratively improve each other. Experiments on two natural language understanding tasks demonstrate that our framework can not only make effective predictions in both supervised and semi-supervised settings, but is also able to generate good natural language explanations. Wangchunshu Zhou, Jinyi Hu, Hanlin Zhang 0002, Xiaodan Liang, Maosong Sun 0001, Chenyan Xiong, Jian Tang 0005 |
NeurIPS | 5 |
| 2020 | Topic-sensitive neural headline generation
Ayana, Lei Xu 0040, Zhiyuan Liu 0001, Maosong Sun 0001 |
Sci. China Inf. Sci. | 5 |
| 2020 | Reinforced Zero-Shot Cross-Lingual Neural Headline GenerationabstractCross-lingual neural headline generation (CNHG), which aims at training a single, large neural network that directly generates a target language headline given a source language news document, has received considerable attention in recent years. Unlike conventional neural headline generation, CNHG faces the problem that there are no large-scale parallel corpora of source language articles and target language headlines. Consequently, CNHG is a zero-shot scenario. To solve this problem, we propose zero resource CNHG with reinforcement learning. We develop a reinforcement learning framework that is composed of two modules: a neural machine translation (NMT) module and a CNHG module. The translation module translates an input document into a source language document, and the headline generation module takes the previous output as input to generate a target language headline. Then, both modules receive a reward for joint training. The experimental results reveal that our method significantly outperforms baseline models. Ayana, Yun Chen 0007, Cheng Yang 0002, Zhiyuan Liu 0001, Maosong Sun 0001 |
IEEE ACM Trans. Audio Speech Lang. Process. | 5 |
| 2020 | Improving Sequence Modeling Ability of Recurrent Neural Networks via SememesabstractSememes, the minimum semantic units of human languages, have been successfully utilized in various natural language processing applications. However, most existing studies exploit sememes in specific tasks and few efforts are made to utilize sememes more fundamentally. In this paper, we propose to incorporate sememes into recurrent neural networks (RNNs) to improve their sequence modeling ability, which is beneficial to all kinds of downstream tasks. We design three different sememe incorporation methods and employ them in typical RNNs including LSTM, GRU and their bidirectional variants. In evaluation, we use several benchmark datasets involving PTB and WikiText-2 for language modeling, SNLI for natural language inference and another two datasets for sentiment analysis and paraphrase detection. Experimental results show evident and consistent improvement of our sememe-incorporated models compared with vanilla RNNs, which proves the effectiveness of our sememe incorporation methods. Moreover, we find the sememe-incorporated models have higher robustness and outperform adversarial training in defending adversarial attack. All the code and data of this work can be obtained at https://github.com/thunlp/SememeRNN. Yujia Qin, Fanchao Qi, Sicong Ouyang, Zhiyuan Liu 0001, Cheng Yang 0002, Yasheng Wang, Qun Liu 0001, Maosong Sun 0001 |
IEEE ACM Trans. Audio Speech Lang. Process. | 8 |
| 2019 | Hybrid Attention-Based Prototypical Networks for Noisy Few-Shot Relation ClassificationabstractThe existing methods for relation classification (RC) primarily rely on distant supervision (DS) because large-scale supervised training datasets are not readily available. Although DS automatically annotates adequate amounts of data for model training, the coverage of this data is still quite limited, and meanwhile many long-tail relations still suffer from data sparsity. Intuitively, people can grasp new knowledge by learning few instances. We thus provide a different view on RC by formalizing RC as a few-shot learning (FSL) problem. However, the current FSL models mainly focus on low-noise vision tasks, which makes them hard to directly deal with the diversity and noise of text. In this paper, we propose hybrid attention-based prototypical networks for the problem of noisy few-shot RC. We design instancelevel and feature-level attention schemes based on prototypical networks to highlight the crucial instances and features respectively, which significantly enhances the performance and robustness of RC models in a noisy FSL scenario. Besides, our attention schemes accelerate the convergence speed of RC models. Experimental results demonstrate that our hybrid attention-based models require fewer training iterations and outperform the state-of-the-art baseline models. The code and datasets are released on https://github.com/thunlp/ HATT-Proto. Tianyu Gao 0001, Xu Han 0007, Zhiyuan Liu 0001, Maosong Sun 0001 |
AAAI | 4 |
| 2019 | Quantifying Similarity between Relations with Fact DistributionabstractWe introduce a conceptually simple and effective method to quantify the similarity between relations in knowledge bases.Specifically, our approach is based on the divergence between the conditional probability distributions over entity pairs.In this paper, these distributions are parameterized by a very simple neural network.Although computing the exact similarity is intractable, we provide a sampling-based method to get a good approximation.We empirically show the outputs of our approach significantly correlate with human judgments.By applying our method to various tasks, we also find that (1) our approach could effectively detect redundant relations extracted by open information extraction (Open IE) models, that (2) even the most competitive models for relational classification still make mistakes among very similar relations, and that (3) our approach could be incorporated into negative sampling and softmax classification to alleviate these mistakes.The source code and experiment details of this paper can be obtained from https://github.com/ thunlp/relation-similarity. Weize Chen, Hao Zhu 0006, Xu Han 0007, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 5 |
| 2019 | XQA: A Cross-lingual Open-domain Question Answering DatasetabstractOpen-domain question answering (OpenQA) aims to answer questions through text retrieval and reading comprehension.Recently, lots of neural network-based models have been proposed and achieved promising results in OpenQA.However, the success of these models relies on a massive volume of training data (usually in English), which is not available in many other languages, especially for those low-resource languages.Therefore, it is essential to investigate cross-lingual OpenQA.In this paper, we construct a novel dataset XQA for cross-lingual OpenQA research.It consists of a training set in English as well as development and test sets in eight other languages.Besides, we provide several baseline systems for cross-lingual OpenQA, including two machine translation-based methods and one zero-shot cross-lingual method (multilingual BERT).Experimental results show that the multilingual BERT model achieves the best results in almost all target languages, while the performance of cross-lingual OpenQA is still much lower than that of English.Our analysis indicates that the performance of cross-lingual OpenQA is related to not only how similar the target language and English are, but also how difficult the question set of the target language is.The XQA dataset is publicly available at http://github.com/thunlp/XQA. Jiahua Liu, Yankai Lin 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 4 |
| 2019 | Modeling Semantic Compositionality with Sememe KnowledgeabstractSemantic compositionality (SC) refers to the phenomenon that the meaning of a complex linguistic unit can be composed of the meanings of its constituents.Most related works focus on using complicated compositionality functions to model SC while few works consider external knowledge in models.In this paper, we verify the effectiveness of sememes, the minimum semantic units of human languages, in modeling SC by a confirmatory experiment.Furthermore, we make the first attempt to incorporate sememe knowledge into SC models, and employ the sememeincorporated models in learning representations of multiword expressions, a typical task of SC.In experiments, we implement our models by incorporating knowledge from a famous sememe knowledge base HowNet and perform both intrinsic and extrinsic evaluations.Experimental results show that our models achieve significant performance boost as compared to the baseline methods without considering sememe knowledge.We further conduct quantitative analysis and case studies to demonstrate the effectiveness of applying sememe knowledge in modeling SC.All the code and data of this paper can be obtained on https: //github.com/thunlp/Sememe-SC. Fanchao Qi, Junjie Huang 0003, Chenghao Yang 0001, Zhiyuan Liu 0001, Xiao Chen 0012, Qun Liu 0001, Maosong Sun 0001 |
ACL (1) | 7 |
| 2019 | Reducing Word Omission Errors in Neural Machine Translation: A Contrastive Learning ApproachabstractWhile neural machine translation (NMT) has achieved remarkable success, NMT systems are prone to make word omission errors.In this work, we propose a contrastive learning approach to reducing word omission errors in NMT.The basic idea is to enable the NMT model to assign a higher probability to a ground-truth translation and a lower probability to an erroneous translation, which is automatically constructed from the ground-truth translation by omitting words.We design different types of negative examples depending on the number of omitted words, word frequency, and part of speech.Experiments on Chinese-to-English, German-to-English, and Russian-to-English translation tasks show that our approach is effective in reducing word omission errors and achieves better translation performance than three baseline methods. Zonghan Yang, Yong Cheng 0003, Yang Liu 0005, Maosong Sun 0001 |
ACL (1) | 4 |
| 2019 | DocRED: A Large-Scale Document-Level Relation Extraction DatasetabstractYuan Yao, Deming Ye, Peng Li, Xu Han, Yankai Lin, Zhenghao Liu, Zhiyuan Liu, Lixin Huang, Jie Zhou, Maosong Sun. Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics. 2019. Yuan Yao 0013, Deming Ye, Peng Li 0030, Xu Han 0007, Yankai Lin 0001, Zhenghao Liu 0001, Zhiyuan Liu 0001, Lixin Huang, Jie Zhou 0016, Maosong Sun 0001 |
ACL (1) | 10 |
| 2019 | ERNIE: Enhanced Language Representation with Informative EntitiesabstractNeural language representation models such as BERT pre-trained on large-scale corpora can well capture rich semantic patterns from plain text, and be fine-tuned to consistently improve the performance of various NLP tasks.However, the existing pre-trained language models rarely consider incorporating knowledge graphs (KGs), which can provide rich structured knowledge facts for better language understanding.We argue that informative entities in KGs can enhance language representation with external knowledge.In this paper, we utilize both large-scale textual corpora and KGs to train an enhanced language representation model (ERNIE), which can take full advantage of lexical, syntactic, and knowledge information simultaneously.The experimental results have demonstrated that ERNIE achieves significant improvements on various knowledge-driven tasks, and meanwhile is comparable with the state-of-the-art model BERT on other common NLP tasks.The source code and experiment details of this paper can be obtained from https:// github.com/thunlp/ERNIE. Zhengyan Zhang, Xu Han 0007, Zhiyuan Liu 0001, Xin Jiang 0002, Maosong Sun 0001, Qun Liu 0001 |
ACL (1) | 5 |
| 2019 | GEAR: Graph-based Evidence Aggregating and Reasoning for Fact VerificationabstractFact verification (FV) is a challenging task which requires to retrieve relevant evidence from plain text and use the evidence to verify given claims.Many claims require to simultaneously integrate and reason over several pieces of evidence for verification.However, previous work employs simple models to extract information from evidence without letting evidence communicate with each other, e.g., merely concatenate the evidence for processing.Therefore, these methods are unable to grasp sufficient relational and logical information among the evidence.To alleviate this issue, we propose a graph-based evidence aggregating and reasoning (GEAR) framework which enables information to transfer on a fully-connected evidence graph and then utilizes different aggregators to collect multievidence information.We further employ BERT, an effective pre-trained language representation model, to improve the performance.Experimental results on a large-scale benchmark dataset FEVER have demonstrated that GEAR could leverage multi-evidence information for FV and thus achieves the promising result with a test FEVER score of 67.10%. Jie Zhou 0024, Xu Han 0007, Cheng Yang 0002, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 7 |
| 2019 | Graph Neural Networks with Generated Parameters for Relation Extractionabstract10.18653/v1/P19-1128 Hao Zhu 0006, Yankai Lin 0001, Zhiyuan Liu 0001, Jie Fu 0001, Tat-Seng Chua, Maosong Sun 0001 |
ACL (1) | 6 |
| 2019 | FewRel 2.0: Towards More Challenging Few-Shot Relation ClassificationabstractTianyu Gao, Xu Han, Hao Zhu, Zhiyuan Liu, Peng Li, Maosong Sun, Jie Zhou. Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP). 2019. Tianyu Gao 0001, Xu Han 0007, Hao Zhu 0006, Zhiyuan Liu 0001, Peng Li 0030, Maosong Sun 0001, Jie Zhou 0016 |
EMNLP/IJCNLP (1) | 6 |
| 2019 | Learning to Copy for Automatic Post-EditingabstractXuancheng Huang, Yang Liu, Huanbo Luan, Jingfang Xu, Maosong Sun. Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP). 2019. Xuancheng Huang, Yang Liu 0005, Huan-Bo Luan, Jingfang Xu, Maosong Sun 0001 |
EMNLP/IJCNLP (1) | 5 |
| 2019 | Improving Back-Translation with Uncertainty-based Confidence EstimationabstractShuo Wang, Yang Liu, Chao Wang, Huanbo Luan, Maosong Sun. Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP). 2019. Shuo Wang 0013, Yang Liu 0005, Chao Wang 0049, Huan-Bo Luan, Maosong Sun 0001 |
EMNLP/IJCNLP (1) | 5 |
| 2019 | HMEAE: Hierarchical Modular Event Argument ExtractionabstractXiaozhi Wang, Ziqi Wang, Xu Han, Zhiyuan Liu, Juanzi Li, Peng Li, Maosong Sun, Jie Zhou, Xiang Ren. Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP). 2019. Xiaozhi Wang, Ziqi Wang 0003, Xu Han 0007, Zhiyuan Liu 0001, Juan-Zi Li, Peng Li 0030, Maosong Sun 0001, Jie Zhou 0016, Xiang Ren 0001 |
EMNLP/IJCNLP (1) | 7 |
| 2019 | Open Relation Extraction: Relational Knowledge Transfer from Supervised Data to Unsupervised DataabstractRuidong Wu, Yuan Yao, Xu Han, Ruobing Xie, Zhiyuan Liu, Fen Lin, Leyu Lin, Maosong Sun. Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP). 2019. Ruidong Wu, Yuan Yao 0013, Xu Han 0007, Ruobing Xie, Zhiyuan Liu 0001, Fen Lin 0002, Leyu Lin, Maosong Sun 0001 |
EMNLP/IJCNLP (1) | 8 |
| 2019 | Sentiment-Controllable Chinese Poetry GenerationabstractExpressing diverse sentiments is one of the main purposes of human poetry creation. Existing Chinese poetry generation models have made great progress in poetry quality, but they all neglected to endow generated poems with specific sentiments. Such defect leads to strong sentiment collapse or bias and thus hurts the diversity and semantics of generated poems. Meanwhile, there are few sentimental Chinese poetry resources for studying. To address this problem, we first collect a manually-labelled sentimental poetry corpus with fine-grained sentiment labels. Then we propose a novel semi-supervised conditional Variational Auto-Encoder model for sentiment-controllable poetry generation. Besides, since poetry is discourse-level text where the polarity and intensity of sentiment could transfer among lines, we incorporate a temporal module to capture sentiment transition patterns among different lines. Experimental results show our model can control the sentiment of not only a whole poem but also each line, and improve the poetry diversity against the state-of-the-art models without losing quality. Xiaoyuan Yi, Maosong Sun 0001, Wenhao Li 0003, Cheng Yang 0002, Zhipeng Guo 0001 |
IJCAI | 3 |
| 2019 | Enhancing Stock Movement Prediction with Adversarial TrainingabstractThis paper contributes a new machine learning solution for stock movement prediction, which aims to predict whether the price of a stock will be up or down in the near future. The key novelty is that we propose to employ adversarial training to improve the generalization of a neural network prediction model. The rationality of adversarial training here is that the input features to stock prediction are typically based on stock price, which is essentially a stochastic variable and continuously changed with time by nature. As such, normal training with static price-based features (e.g. the close price) can easily overfit the data, being insufficient to obtain reliable models. To address this problem, we propose to add perturbations to simulate the stochasticity of price variable, and train the model to work well under small yet intentional perturbations. Extensive experiments on two real-world stock data show that our method outperforms the state-of-the-art solution [Xu and Cohen, 2018] with 3.11% relative improvements on average w.r.t. accuracy, validating the usefulness of adversarial training for stock prediction task. Fuli Feng, Xiangnan He 0001, Ji Ding 0005, Maosong Sun 0001, Tat-Seng Chua |
IJCAI | 5 |
| 2019 | Multi-scale Information Diffusion Prediction with Reinforced Recurrent NetworksabstractInformation diffusion prediction is an important task which studies how information items spread among users. With the success of deep learning techniques, recurrent neural networks (RNNs) have shown their powerful capability in modeling information diffusion as sequential data. However, previous works focused on either microscopic diffusion prediction which aims at guessing the next influenced user or macroscopic diffusion prediction which estimates the total numbers of influenced users during the diffusion process. To the best of our knowledge, no previous works have suggested a unified model for both microscopic and macroscopic scales. In this paper, we propose a novel multi-scale diffusion prediction model based on reinforcement learning (RL). RL incorporates the macroscopic diffusion size information into the RNN-based microscopic diffusion model by addressing the non-differentiable problem. We also employ an effective structural context extraction strategy to utilize the underlying social graph information. Experimental results show that our proposed model outperforms state-of-the-art baseline models on both microscopic and macroscopic diffusion predictions on three real-world datasets. Cheng Yang 0002, Jian Tang 0005, Maosong Sun 0001, Ganqu Cui, Zhiyuan Liu 0001 |
IJCAI | 3 |
| 2019 | Multi-Round Transfer Learning for Low-Resource NMT Using Multiple High-Resource LanguagesabstractNeural machine translation (NMT) has made remarkable progress in recent years, but the performance of NMT suffers from a data sparsity problem since large-scale parallel corpora are only readily available for high-resource languages (HRLs). In recent days, transfer learning (TL) has been used widely in low-resource languages (LRLs) machine translation, while TL is becoming one of the vital directions for addressing the data sparsity problem in low-resource NMT. As a solution, a transfer learning method in NMT is generally obtained via initializing the low-resource model (child) with the high-resource model (parent). However, leveraging the original TL to low-resource models is neither able to make full use of highly related multiple HRLs nor to receive different parameters from the same parents. In order to exploit multiple HRLs effectively, we present a language-independent and straightforward multi-round transfer learning (MRTL) approach to low-resource NMT. Besides, with the intention of reducing the differences between high-resource and low-resource languages at the character level, we introduce a unified transliteration method for various language families, which are both semantically and syntactically highly analogous with each other. Experiments on low-resource datasets show that our approaches are effective, significantly outperform the state-of-the-art methods, and yield improvements of up to 5.63 BLEU points. Mieradilijiang Maimaiti, Yang Liu 0005, Huan-Bo Luan, Maosong Sun 0001 |
ACM Trans. Asian Low Resour. Lang. Inf. Process. | 4 |
| 2019 | A Unified Framework for Community Detection and Network Representation LearningabstractNetwork representation learning (NRL) aims to learn low-dimensional vectors for vertices in a network. Most existing NRL methods focus on learning representations from local context of vertices (such as their neighbors). Nevertheless, vertices in many complex networks also exhibit significant global patterns widely known as communities. It's intuitive that vertices in the same community tend to connect densely and share common attributes. These patterns are expected to improve NRL and benefit relevant evaluation tasks, such as link prediction and vertex classification. Inspired by the analogy between network representation learning and text modeling, we propose a unified NRL framework by introducing community information of vertices, named as Community-enhanced Network Representation Learning (CNRL). CNRL simultaneously detects community distribution of each vertex and learns embeddings of both vertices and communities. Moreover, the proposed community enhancement mechanism can be applied to various existing NRL models. In experiments, we evaluate our model on vertex classification, link prediction, and community detection using several real-world datasets. The results demonstrate that CNRL significantly and consistently outperforms other state-of-the-art methods while verifying our assumptions on the correlations between vertices and communities. Cunchao Tu, Xiangkai Zeng, Hao Wang 0214, Zhengyan Zhang, Zhiyuan Liu 0001, Maosong Sun 0001, Bo Zhang 0056, Leyu Lin |
IEEE Trans. Knowl. Data Eng. | 6 |
| 2018 | Neural Knowledge Acquisition via Mutual Attention Between Knowledge Graph and TextabstractWe propose a general joint representation learning framework for knowledge acquisition (KA) on two tasks, knowledge graph completion (KGC) and relation extraction (RE) from text. In this framework, we learn representations of knowledge graphs (KGs) and text within a unified parameter sharing semantic space. To achieve better fusion, we propose an effective mutual attention between KGs and text. The reciprocal attention mechanism enables us to highlight important features and perform better KGC and RE. Different from conventional joint models, no complicated linguistic analysis or strict alignments between KGs and text are required to train our models. Experiments on relation extraction and entity link prediction show that models trained under our joint framework are significantly improved in comparison with other baselines. Most existing methods for KGC and RE can be easily integrated into our framework due to its flexible architectures. The source code of this paper can be obtained from https://github.com/thunlp/JointNRE. Xu Han 0007, Zhiyuan Liu 0001, Maosong Sun 0001 |
AAAI | 3 |
| 2018 | Improving Neural Fine-Grained Entity Typing With Knowledge AttentionabstractFine-grained entity typing aims to identify the semantic type of an entity in a particular plain text. It is an important task which can be helpful for a lot of natural language processing (NLP) applications. Most existing methods typically extract features separately from the entity mention and context words for type classification. These methods inevitably fail to model complex correlations between entity mentions and context words. They also neglect rich background information about these entities in knowledge bases (KBs). To address these issues, we take information from KBs into consideration to bridge entity mentions and their context together, and thereby propose Knowledge-Attention Neural Fine-Grained Entity Typing. Experimental results and case studies on real-world datasets demonstrate that our model significantly outperforms other state-of-the-art methods, revealing the effectiveness of incorporating KB information for entity typing. Code and data for this paper can be found at https://github.com/thunlp/KNET. Ji Xin, Yankai Lin 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
AAAI | 4 |
| 2018 | Chinese LIWC Lexicon Expansion via Hierarchical Classification of Word Embeddings with Sememe AttentionabstractLinguistic Inquiry and Word Count (LIWC) is a word counting software tool which has been used for quantitative text analysis in many fields. Due to its success and popularity, the core lexicon has been translated into Chinese and many other languages. However, the lexicon only contains several thousand of words, which is deficient compared with the number of common words in Chinese. Current approaches often require manually expanding the lexicon, but it often takes too much time and requires linguistic experts to extend the lexicon. To address this issue, we propose to expand the LIWC lexicon automatically. Specifically, we consider it as a hierarchical classification problem and utilize the Sequence-to-Sequence model to classify words in the lexicon. Moreover, we use the sememe information with the attention mechanism to capture the exact meanings of a word, so that we can expand a more precise and comprehensive lexicon. The experimental results show that our model has a better understanding of word meanings with the help of sememes and achieves significant and consistent improvements compared with the state-of-the-art methods. The source code of this paper can be obtained from https://github.com/thunlp/Auto_CLIWC. Xiangkai Zeng, Cheng Yang 0002, Cunchao Tu, Zhiyuan Liu 0001, Maosong Sun 0001 |
AAAI | 5 |
| 2018 | Denoising Distantly Supervised Open-Domain Question AnsweringabstractDistantly supervised open-domain question answering (DS-QA) aims to find answers in collections of unlabeled text.Existing DS-QA models usually retrieve related paragraphs from a large-scale corpus and apply reading comprehension technique to extract answers from the most relevant paragraph.They ignore the rich information contained in other paragraphs.Moreover, distant supervision data inevitably accompanies with the wrong labeling problem, and these noisy data will substantially degrade the performance of DS-QA.To address these issues, we propose a novel DS-QA model which employs a paragraph selector to filter out those noisy paragraphs and a paragraph reader to extract the correct answer from those denoised paragraphs.Experimental results on real-world datasets show that our model can capture useful information from noisy data and achieve significant improvements on DS-QA as compared to all baselines.The source code and data of this paper can be obtained from https: //github.com/thunlp/OpenQA Yankai Lin 0001, Haozhe Ji, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 4 |
| 2018 | Incorporating Chinese Characters of Words for Lexical Sememe PredictionabstractSememes are minimum semantic units of concepts in human languages, such that each word sense is composed of one or multiple sememes.Words are usually manually annotated with their sememes by linguists, and form linguistic commonsense knowledge bases widely used in various NLP tasks.Recently, the lexical sememe prediction task has been introduced.It consists of automatically recommending sememes for words, which is expected to improve annotation efficiency and consistency.However, existing methods of lexical sememe prediction typically rely on the external context of words to represent the meaning, which usually fails to deal with low-frequency and out-ofvocabulary words.To address this issue for Chinese, we propose a novel framework to take advantage of both internal character information and external context information of words.We experiment on HowNet, a Chinese sememe knowledge base, and demonstrate that our framework outperforms state-of-the-art baselines by a large margin, and maintains a robust performance even for low-frequency words.i Huiming Jin, Hao Zhu 0006, Zhiyuan Liu 0001, Ruobing Xie, Maosong Sun 0001, Fen Lin 0002, Leyu Lin |
ACL (1) | 5 |
| 2018 | Entity-Duet Neural Ranking: Understanding the Role of Knowledge Graph Semantics in Neural Information RetrievalabstractThis paper presents the Entity-Duet Neural Ranking Model (EDRM), which introduces knowledge graphs to neural search systems.EDRM represents queries and documents by their words and entity annotations.The semantics from knowledge graphs are integrated in the distributed representations of their entities, while the ranking is conducted by interaction-based neural ranking networks.The two components are learned end-to-end, making EDRM a natural combination of entityoriented search and neural information retrieval.Our experiments on a commercial search log demonstrate the effectiveness of EDRM.Our analyses reveal that knowledge graph semantics significantly improve the generalization ability of neural ranking models. Zhenghao Liu 0001, Chenyan Xiong, Maosong Sun 0001, Zhiyuan Liu 0001 |
ACL (1) | 3 |
| 2018 | Few-Shot Charge Prediction with Discriminative Legal AttributesabstractAutomatic charge prediction aims to predict the final charges according to the fact descriptions in criminal cases and plays a crucial role in legal assistant systems. Existing works on charge prediction perform adequately on those high-frequency charges but are not yet capable of predicting few-shot charges with limited cases. Moreover, these exist many confusing charge pairs, whose fact descriptions are fairly similar to each other. To address these issues, we introduce several discriminative attributes of charges as the internal mapping between fact descriptions and charges. These attributes provide additional information for few-shot charges, as well as effective signals for distinguishing confusing charges. More specifically, we propose an attribute-attentive charge prediction model to infer the attributes and charges simultaneously. Experimental results on real-work datasets demonstrate that our proposed model achieves significant and consistent improvements than other state-of-the-art baselines. Specifically, our model outperforms other baselines by more than 50% in the few-shot scenario. Our codes and datasets can be obtained from https://github.com/thunlp/attribute_charge. Zikun Hu, Xiang Li 0105, Cunchao Tu, Zhiyuan Liu 0001, Maosong Sun 0001 |
COLING | 5 |
| 2018 | Adversarial Multi-lingual Neural Relation ExtractionabstractMulti-lingual relation extraction aims to find unknown relational facts from text in various languages. Existing models cannot well capture the consistency and diversity of relation patterns in different languages. To address these issues, we propose an adversarial multi-lingual neural relation extraction (AMNRE) model, which builds both consistent and individual representations for each sentence to consider the consistency and diversity among languages. Further, we adopt an adversarial training strategy to ensure those consistent sentence representations could effectively extract the language-consistent relation patterns. The experimental results on real-world datasets demonstrate that our AMNRE model significantly outperforms the state-of-the-art models. The source code of this paper can be obtained from https://github.com/thunlp/AMNRE. Xiaozhi Wang, Xu Han 0007, Yankai Lin 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
COLING | 5 |
| 2018 | Chinese Poetry Generation with a Salient-Clue MechanismabstractAs a precious part of the human cultural heritage, Chinese poetry has influenced people for generations.Automatic poetry composition is a challenge for AI.In recent years, significant progress has been made in this area benefiting from the development of neural networks.However, the coherence in meaning, theme or even artistic conception for a generated poem as a whole still remains a big problem.In this paper, we propose a novel Salient-Clue mechanism for Chinese poetry generation.Different from previous work which tried to exploit all the context information, our model selects the most salient characters automatically from each so-far generated line to gradually form a salient clue, which is utilized to guide successive poem generation process so as to eliminate interruptions and improve coherence.Besides, our model can be flexibly extended to control the generated poem in different aspects, for example, poetry style, which further enhances the coherence.Experimental results show that our model is very effective, outperforming three strong baselines. Xiaoyuan Yi, Maosong Sun 0001 |
CoNLL | 3 |
| 2018 | Language Modeling with Sparse Product of Sememe ExpertsabstractMost language modeling methods rely on large-scale data to statistically learn the sequential patterns of words.In this paper, we argue that words are atomic language units but not necessarily atomic semantic units.Inspired by HowNet, we use sememes, the minimum semantic units in human languages, to represent the implicit semantics behind words for language modeling, named Sememe-Driven Language Model (SDLM).More specifically, to predict the next word, SDLM first estimates the sememe distribution given textual context.Afterwards, it regards each sememe as a distinct semantic expert, and these experts jointly identify the most probable senses and the corresponding word.In this way, SDLM enables language models to work beyond word-level manipulation to fine-grained sememe-level semantics, and offers us more powerful tools to fine-tune language models and improve the interpretability as well as the robustness of language models.Experiments on language modeling and the downstream application of headline generation demonstrate the significant effectiveness of SDLM. Yihong Gu, Jun Yan 0012, Hao Zhu 0006, Zhiyuan Liu 0001, Ruobing Xie, Maosong Sun 0001, Fen Lin 0002, Leyu Lin |
EMNLP | 6 |
| 2018 | Hierarchical Relation Extraction with Coarse-to-Fine Grained AttentionabstractDistantly supervised relation extraction employs existing knowledge graphs to automatically collect training data.While distant supervision is effective to scale relation extraction up to large-scale corpora, it inevitably suffers from the wrong labeling problem.Many efforts have been devoted to identifying valid instances from noisy data.However, most existing methods handle each relation in isolation, regardless of rich semantic correlations located in relation hierarchies.In this paper, we aim to incorporate the hierarchical information of relations for distantly supervised relation extraction and propose a novel hierarchical attention scheme.The multiple layers of our hierarchical attention scheme provide coarseto-fine granularity to better identify valid instances, which is especially effective for extracting those long-tail relations.The experimental results on a large-scale benchmark dataset demonstrate that our models are capable of modeling the hierarchical information of relations and significantly outperform other baselines.The source code of this paper can be obtained from https://github.com/ thunlp/HNRE. Xu Han 0007, Pengfei Yu 0001, Zhiyuan Liu 0001, Maosong Sun 0001, Peng Li 0030 |
EMNLP | 4 |
| 2018 | FewRel: A Large-Scale Supervised Few-shot Relation Classification Dataset with State-of-the-Art EvaluationabstractWe present a Few-Shot Relation Classification Dataset (FewRel), consisting of 70, 000 sentences on 100 relations derived from Wikipedia and annotated by crowdworkers.The relation of each sentence is first recognized by distant supervision methods, and then filtered by crowdworkers.We adapt the most recent state-of-the-art few-shot learning methods for relation classification and conduct thorough evaluation of these methods.Empirical results show that even the most competitive few-shot learning models struggle on this task, especially as compared with humans.We also show that a range of different reasoning skills are needed to solve our task.These results indicate that few-shot relation classification remains an open problem and still requires further research.Our detailed analysis points multiple directions for future research.All details and resources about the dataset and baselines are released on http://zhuhao.me/ fewrel. Xu Han 0007, Hao Zhu 0006, Pengfei Yu 0001, Yuan Yao 0013, Zhiyuan Liu 0001, Maosong Sun 0001 |
EMNLP | 7 |
| 2018 | A Multi-answer Multi-task Framework for Real-world Machine Reading ComprehensionabstractThe task of machine reading comprehension (MRC) has evolved from answering simple questions from well-edited text to answering real questions from users out of web data.In the real-world setting, full-body text from multiple relevant documents in the top search results are provided as context for questions from user queries, including not only questions with a single, short, and factual answer, but also questions about reasons, procedures, and opinions.In this case, multiple answers could be equally valid for a single question and each answer may occur multiple times in the context, which should be taken into consideration when we build MRC system.We propose a multi-answer multi-task framework, in which different loss functions are used for multiple reference answers.Minimum Risk Training is applied to solve the multi-occurrence problem of a single answer.Combined with a simple heuristic passage extraction strategy for overlong documents, our model increases the ROUGE-L score on the DuReader dataset from 44.18, the previous state-of-the-art, to 51.09. Jiahua Liu, Wan Wei, Maosong Sun 0001, Yantao Du, Dekang Lin |
EMNLP | 3 |
| 2018 | Cross-lingual Lexical Sememe PredictionabstractSememes are defined as the minimum semantic units of human languages.As important knowledge sources, sememe-based linguistic knowledge bases have been widely used in many NLP tasks.However, most languages still do not have sememe-based linguistic knowledge bases.Thus we present a task of cross-lingual lexical sememe prediction, aiming to automatically predict sememes for words in other languages.We propose a novel framework to model correlations between sememes and multi-lingual words in low-dimensional semantic space for sememe prediction.Experimental results on real-world datasets show that our proposed model achieves consistent and significant improvements as compared to baseline methods in cross-lingual sememe prediction.The codes and data of this paper are available at https: //github.com/thunlp/CL-SP. Fanchao Qi, Yankai Lin 0001, Maosong Sun 0001, Hao Zhu 0006, Ruobing Xie, Zhiyuan Liu 0001 |
EMNLP | 3 |
| 2018 | Put It Back: Entity Typing with Language Model EnhancementabstractEntity typing aims to classify semantic types of an entity mention in a specific context.Most existing models obtain training data using distant supervision, and inevitably suffer from the problem of noisy labels.To address this issue, we propose entity typing with language model enhancement.It utilizes a language model to measure the compatibility between context sentences and labels, and thereby automatically focuses more on context-dependent labels.Experiments on benchmark datasets demonstrate that our method is capable of enhancing the entity typing model with information from the language model, and significantly outperforms the stateof-the-art baseline. Ji Xin, Hao Zhu 0006, Xu Han 0007, Zhiyuan Liu 0001, Maosong Sun 0001 |
EMNLP | 5 |
| 2018 | Stylistic Chinese Poetry Generation via Unsupervised Style DisentanglementabstractThe ability to write diverse poems in different styles under the same poetic imagery is an important characteristic of human poetry writing.Most previous works on automatic Chinese poetry generation focused on improving the coherency among lines.Some work explored style transfer but suffered from expensive expert labeling of poem styles.In this paper, we target on stylistic poetry generation in a fully unsupervised manner for the first time.We propose a novel model which requires no supervised style labeling by incorporating mutual information, a concept in information theory, into modeling.Experimental results show that our model is able to generate stylistic poems without losing fluency and coherency. Cheng Yang 0002, Maosong Sun 0001, Xiaoyuan Yi, Wenhao Li 0003 |
EMNLP | 2 |
| 2018 | Automatic Poetry Generation with Mutual Reinforcement LearningabstractPoetry is one of the most beautiful forms of human language art.As a crucial step towards computer creativity, automatic poetry generation has drawn researchers' attention for decades.In recent years, some neural models have made remarkable progress in this task.However, they are all based on maximum likelihood estimation, which only learns common patterns of the corpus and results in lossevaluation mismatch.Human experts evaluate poetry in terms of some specific criteria, instead of word-level likelihood.To handle this problem, we directly model the criteria and use them as explicit rewards to guide gradient update by reinforcement learning, so as to motivate the model to pursue higher scores.Besides, inspired by writing theories, we propose a novel mutual reinforcement learning schema.We simultaneously train two learners (generators) which learn not only from the teacher (rewarder) but also from each other to further improve performance.We experiment on Chinese poetry.Based on a strong basic model, our method achieves better results and outperforms the current state-of-theart method. Xiaoyuan Yi, Maosong Sun 0001, Wenhao Li 0003 |
EMNLP | 2 |
| 2018 | Improving the Transformer Translation Model with Document-Level ContextabstractAlthough the Transformer translation model (Vaswani et al., 2017) has achieved state-ofthe-art performance in a variety of translation tasks, how to use document-level context to deal with discourse phenomena problematic for Transformer still remains a challenge.In this work, we extend the Transformer model with a new context encoder to represent document-level context, which is then incorporated into the original encoder and decoder.As large-scale document-level parallel corpora are usually not available, we introduce a two-step training method to take full advantage of abundant sentence-level parallel corpora and limited document-level parallel corpora.Experiments on the NIST Chinese-English datasets and the IWSLT French-English datasets show that our approach improves over Transformer significantly. 1 Huan-Bo Luan, Maosong Sun 0001, Feifei Zhai, Jingfang Xu, Min Zhang 0005, Yang Liu 0005 |
EMNLP | 3 |
| 2018 | Legal Judgment Prediction via Topological LearningabstractLegal Judgment Prediction (LJP) aims to predict the judgment result based on the facts of a case and becomes a promising application of artificial intelligence techniques in the legal field.In real-world scenarios, legal judgment usually consists of multiple subtasks, such as the decisions of applicable law articles, charges, fines, and the term of penalty.Moreover, there exist topological dependencies among these subtasks.While most existing works only focus on a specific subtask of judgment prediction and ignore the dependencies among subtasks, we formalize the dependencies among subtasks as a Directed Acyclic Graph (DAG) and propose a topological multi-task learning framework, TOP-JUDGE, which incorporates multiple subtasks and DAG dependencies into judgment prediction.We conduct experiments on several realworld large-scale datasets of criminal cases in the civil law system.Experimental results show that our model achieves consistent and significant improvements over baselines on all judgment prediction tasks.The source code can be obtained from https://github. com/thunlp/TopJudge. Haoxi Zhong, Zhipeng Guo 0001, Cunchao Tu, Chaojun Xiao, Zhiyuan Liu 0001, Maosong Sun 0001 |
EMNLP | 6 |
| 2018 | Chinese Poetry Generation with a Working Memory ModelabstractAs an exquisite and concise literary form, poetry is a gem of human culture. Automatic poetry generation is an essential step towards computer creativity. In recent years, several neural models have been designed for this task. However, among lines of a whole poem, the coherence in meaning and topics still remains a big challenge. In this paper, inspired by the theoretical concept in cognitive psychology, we propose a novel Working Memory model for poetry generation. Different from previous methods, our model explicitly maintains topics and informative limited history in a neural memory. During the generation process, our model reads the most relevant parts from memory slots to generate the current line. After each line is generated, it writes the most salient parts of the previous line into memory slots. By dynamic manipulation of the memory, our model keeps a coherent information flow and learns to express each topic flexibly and naturally. We experiment on three different genres of Chinese poetry: quatrain, iambic and chinoiserie lyric. Both automatic and human evaluation results show that our model outperforms current state-of-the-art methods. Xiaoyuan Yi, Maosong Sun 0001, Zonghan Yang |
IJCAI | 2 |
| 2018 | Bandit Learning with Implicit FeedbackabstractImplicit feedback, such as user clicks, although abundant in online information service systems, does not provide substantial evidence on users' evaluation of system's output. Without proper modeling, such incomplete supervision inevitably misleads model estimation, especially in a bandit learning setting where the feedback is acquired on the fly. In this work, we perform contextual bandit learning with implicit feedback by modeling the feedback as a composition of user result examination and relevance judgment. Since users' examination behavior is unobserved, we introduce latent variables to model it. We perform Thompson sampling on top of variational Bayesian inference for arm selection and model update. Our upper regret bound analysis of the proposed algorithm proves its feasibility of learning from implicit feedback in a bandit setting; and extensive empirical evaluations on click logs collected from a major MOOC platform further demonstrate its learning effectiveness in practice. Qingyun Wu, Hongning Wang, Jie Tang 0001, Maosong Sun 0001 |
NeurIPS | 5 |
| 2018 | Query Suggestion with Feedback Memory NetworkabstractThis paper presents Feedback Memory Network (\textttFMN) which models user interactions with the search engine for query suggestion. Besides modeling the queries issued by the user, \textttFMN also considers user feedback on the search results. It converts user browsing and click actions to the attention over the top-ranked documents and combines them into the feedback memories of the query, thus better models the underlying information needs. The feedback memories and the query sequence are then combined to suggest queries by the sequence-to-sequence neural network. Modeling user feedback makes it possible to suggest diverse queries for the same query sequence, if users have preferred different search results that indicate different information needs. Our experiments on the search log from a Chinese commercial search engine showed the stable and robust advantages of \textttFMN. Especially when the feedback is richer or more informative, \textttFMN provides more diverse and accurate suggestions, which is exceptionally helpful for ambiguous sessions where more information is required to infer the search intents. Chenyan Xiong, Maosong Sun 0001, Zhiyuan Liu 0001 |
WWW | 3 |
| 2018 | Zero-Shot Cross-Lingual Neural Headline GenerationabstractNeural headline generation (NHG) has been proven to be effective in generating a fully abstractive headline recently. Existing NHG systems are only capable of producing headline of the same language as the original document. Cross lingual headline generation is an important task since it provides an efficient way to understand the key point of a document in a different language. Due to the lack of those parallel corpora of direct source language articles and target language headlines, we propose to deal with the cross-lingual neural headline generation (CNHG) under the zero-shot scenario. A trivial solution is to translate and summarize the source document in a pipeline way. However, a pipeline solution will lead to error propagation in the translation and summarization phases. This challenge motivates us to build a direct source-to-target CNHG model based on existing parallel corpora of translation and monolingual headline generation. Specifically, we let a parameterized CNHG model (student model) mimic the output of a pretrained translation or headline generation model (teacher model). To the best of our knowledge, this is the first effort to address CNHG problem. Besides, we construct English-Chinese headline generation evaluation datasets by manual translation. Experimental results on English-to-Chinese cross-lingual headline generation demonstrate that our proposed method significantly outperforms the baseline models. Shiqi Shen, Yun Chen 0007, Cheng Yang 0002, Zhiyuan Liu 0001, Maosong Sun 0001 |
IEEE ACM Trans. Audio Speech Lang. Process. | 5 |
| 2018 | Joint POS Tagging and Dependence Parsing With Transition-Based Neural NetworksabstractWhile part-of-speech (POS) tagging and dependency parsing are observed to be closely related, existing work on joint modeling with manually crafted feature templates suffers from the feature sparsity and incompleteness problems. In this paper, we propose an approach to joint POS tagging and dependency parsing using transition-based neural networks. Three neural network based classifiers are designed to resolve shift/reduce, tagging, and labeling conflicts. Experiments show that our approach significantly outperforms previous methods for joint POS tagging and dependency parsing across a variety of natural languages. Liner Yang, Meishan Zhang, Yang Liu 0005, Maosong Sun 0001, Guohong Fu |
IEEE ACM Trans. Audio Speech Lang. Process. | 4 |
| 2017 | Bilingual Lexicon Induction from Non-Parallel Data with Minimal SupervisionabstractBuilding bilingual lexica from non-parallel data is a long-standing natural language processing research problem that could benefit thousands of resource-scarce languages which lack parallel data. Recent advances of continuous word representations have opened up new possibilities for this task, e.g. by establishing cross-lingual mapping between word embeddings via a seed lexicon. The method is however unreliable when there are only a limited number of seeds, which is a reasonable setting for resource-scarce languages. We tackle the limitation by introducing a novel matching mechanism into bilingual word representation learning. It captures extra translation pairs exposed by the seeds to incrementally improve the bilingual word embeddings. In our experiments, we find the matching mechanism to substantially improve the quality of the bilingual vector space, which in turn allows us to induce better bilingual lexica with seeds as few as 10. Meng Zhang 0019, Haoruo Peng, Yang Liu 0005, Huan-Bo Luan, Maosong Sun 0001 |
AAAI | 5 |
| 2017 | Visualizing and Understanding Neural Machine TranslationabstractWhile neural machine translation (NMT) has made remarkable progress in recent years, it is hard to interpret its internal workings due to the continuous representations and non-linearity of neural networks.In this work, we propose to use layer-wise relevance propagation (LRP) to compute the contribution of each contextual word to arbitrary hidden states in the attention-based encoderdecoder framework.We show that visualization with LRP helps to interpret the internal workings of NMT and analyze translation errors. Yanzhuo Ding, Yang Liu 0005, Huan-Bo Luan, Maosong Sun 0001 |
ACL (1) | 4 |
| 2017 | Neural Relation Extraction with Multi-lingual AttentionabstractRelation extraction has been widely used for finding unknown relational facts from the plain text.Most existing methods focus on exploiting mono-lingual data for relation extraction, ignoring massive information from the texts in various languages.To address this issue, we introduce a multi-lingual neural relation extraction framework, which employs monolingual attention to utilize the information within mono-lingual texts and further proposes cross-lingual attention to consider the information consistency and complementarity among cross-lingual texts.Experimental results on real-world datasets show that our model can take advantage of multi-lingual texts and consistently achieve significant improvements on relation extraction as compared with baselines.The source code of this paper can be obtained from https://github.com/thunlp/MNRE Yankai Lin 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 3 |
| 2017 | Improved Word Representation Learning with SememesabstractSememes are minimum semantic units of word meanings, and the meaning of each word sense is typically composed by several sememes.Since sememes are not explicit for each word, people manually annotate word sememes and form linguistic common-sense knowledge bases.In this paper, we present that, word sememe information can improve word representation learning (WRL), which maps words into a low-dimensional semantic space and serves as a fundamental step for many NLP tasks.The key idea is to utilize word sememes to capture exact meanings of a word within specific contexts accurately.More specifically, we follow the framework of Skip-gram and present three sememe-encoded models to learn representations of sememes, senses and words, where we apply the attention scheme to detect word senses in various contexts.We conduct experiments on two tasks including word similarity and word analogy, and our models significantly outperform baselines.The results indicate that WRL can benefit from sememes via the attention scheme, and also confirm our models being capable of correctly modeling sememe information. Yilin Niu, Ruobing Xie, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 4 |
| 2017 | CANE: Context-Aware Network Embedding for Relation ModelingabstractNetwork embedding (NE) is playing a critical role in network analysis, due to its ability to represent vertices with efficient low-dimensional embedding vectors.However, existing NE models aim to learn a fixed context-free embedding for each vertex and neglect the diverse roles when interacting with other vertices.In this paper, we assume that one vertex usually shows different aspects when interacting with different neighbor vertices, and should own different embeddings respectively.Therefore, we present Context-Aware Network Embedding (CANE), a novel NE model to address this issue.CANE learns context-aware embeddings for vertices with mutual attention mechanism and is expected to model the semantic relationships between vertices more precisely.In experiments, we compare our model with existing NE models on three real-world datasets.Experimental results show that CANE achieves significant improvement than state-of-the-art methods on link prediction and comparable performance on vertex classification.The source code and datasets can be obtained from https://github.com/thunlp/CANE. Cunchao Tu, Zhiyuan Liu 0001, Maosong Sun 0001 |
ACL (1) | 4 |
| 2017 | Adversarial Training for Unsupervised Bilingual Lexicon InductionabstractWord embeddings are well known to capture linguistic regularities of the language on which they are trained.Researchers also observe that these regularities can transfer across languages.However, previous endeavors to connect separate monolingual word embeddings typically require cross-lingual signals as supervision, either in the form of parallel corpus or seed lexicon.In this work, we show that such cross-lingual connection can actually be established without any form of supervision.We achieve this end by formulating the problem as a natural adversarial game, and investigating techniques that are crucial to successful training.We carry out evaluation on the unsupervised bilingual lexicon induction task.Even though this task appears intrinsically cross-lingual, we are able to demonstrate encouraging performance without any cross-lingual clues. Meng Zhang 0019, Yang Liu 0005, Huan-Bo Luan, Maosong Sun 0001 |
ACL (1) | 4 |
| 2017 | Prior Knowledge Integration for Neural Machine Translation using Posterior RegularizationabstractAlthough neural machine translation has made significant progress recently, how to integrate multiple overlapping, arbitrary prior knowledge sources remains a challenge.In this work, we propose to use posterior regularization to provide a general framework for integrating prior knowledge into neural machine translation.We represent prior knowledge sources as features in a log-linear model, which guides the learning process of the neural translation model.Experiments on Chinese-English translation show that our approach leads to significant improvements. Yang Liu 0005, Huan-Bo Luan, Jingfang Xu, Maosong Sun 0001 |
ACL (1) | 5 |
| 2017 | Incorporating Relation Paths in Neural Relation ExtractionabstractDistantly supervised relation extraction has been widely used to find novel relational facts from plain text.To predict the relation between a pair of two target entities, existing methods solely rely on those direct sentences containing both entities.In fact, there are also many sentences containing only one of the target entities, which also provide rich useful information but not yet employed by relation extraction.To address this issue, we build inference chains between two target entities via intermediate entities, and propose a path-based neural relation extraction model to encode the relational semantics from both direct sentences and inference chains.Experimental results on realworld datasets show that, our model can make full use of those sentences containing only one target entity, and achieves significant and consistent improvements on relation extraction as compared with strong baselines.The source code of this paper can be obtained from https:// github.com/thunlp/PathNRE. Wenyuan Zeng, Yankai Lin 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
EMNLP | 4 |
| 2017 | Earth Mover's Distance Minimization for Unsupervised Bilingual Lexicon InductionabstractCross-lingual natural language processing hinges on the premise that there exists invariance across languages. At the word level, researchers have identified such invariance in the word embedding semantic spaces of different languages. However, in order to connect the separate spaces, cross-lingual supervision encoded in parallel data is typically required. In this paper, we attempt to establish the cross-lingual connection without relying on any cross-lingual supervision. By viewing word embedding spaces as distributions, we propose to minimize their earth mover's distance, a measure of divergence between distributions. We demonstrate the success on the unsupervised bilingual lexicon induction task. In addition, we reveal an interesting finding that the earth mover's distance shows potential as a measure of language difference. Meng Zhang 0019, Yang Liu 0005, Huan-Bo Luan, Maosong Sun 0001 |
EMNLP | 4 |
| 2017 | Joint Training for Pivot-based Neural Machine TranslationabstractWhile recent neural machine translation approaches have delivered state-of-the-art performance for resource-rich language pairs, they suffer from the data scarcity problem for resource-scarce language pairs. Although this problem can be alleviated by exploiting a pivot language to bridge the source and target languages, the source-to-pivot and pivot-to-target translation models are usually independently trained. In this work, we introduce a joint training algorithm for pivot-based neural machine translation. We propose three methods to connect the two models and enable them to interact with each other during training. Experiments on Europarl and WMT corpora show that joint training of source-to-pivot and pivot-to-target models leads to significant improvements over independent training across various languages. Yong Cheng 0003, Qian Yang 0003, Yang Liu 0005, Maosong Sun 0001, Wei Xu 0005 |
IJCAI | 4 |
| 2017 | TransNet: Translation-Based Network Representation Learning for Social Relation ExtractionabstractConventional network representation learning (NRL) models learn low-dimensional vertex representations by simply regarding each edge as a binary or continuous value. However, there exists rich semantic information on edges and the interactions between vertices usually preserve distinct meanings, which are largely neglected by most existing NRL models. In this work, we present a novel Translation-based NRL model, TransNet, by regarding the interactions between vertices as a translation operation. Moreover, we formalize the task of Social Relation Extraction (SRE) to evaluate the capability of NRL methods on modeling the relations between vertices. Experimental results on SRE demonstrate that TransNet significantly outperforms other baseline methods by 10% to 20% on hits@1. The source code and datasets can be obtained from https://github.com/thunlp/TransNet. Cunchao Tu, Zhengyan Zhang, Zhiyuan Liu 0001, Maosong Sun 0001 |
IJCAI | 4 |
| 2017 | Image-embodied Knowledge Representation LearningabstractEntity images could provide significant visual information for knowledge representation learning. Most conventional methods learn knowledge representations merely from structured triples, ignoring rich visual information extracted from entity images. In this paper, we propose a novel Image-embodied Knowledge Representation Learning model (IKRL), where knowledge representations are learned with both triple facts and images. More specifically, we first construct representations for all images of an entity with a neural image encoder. These image representations are then integrated into an aggregated image-based representation via an attention-based method. We evaluate our IKRL models on knowledge graph completion and triple classification. Experimental results demonstrate that our models outperform all baselines on both tasks, which indicates the significance of visual information for knowledge representations and the capability of our models in learning knowledge representations with images. Ruobing Xie, Zhiyuan Liu 0001, Huan-Bo Luan, Maosong Sun 0001 |
IJCAI | 4 |
| 2017 | Lexical Sememe Prediction via Word Embeddings and Matrix FactorizationabstractSememes are defined as the minimum semantic units of human languages. People have manually annotated lexical sememes for words and form linguistic knowledge bases. However, manual construction is time-consuming and labor-intensive, with significant annotation inconsistency and noise. In this paper, we for the first time explore to automatically predict lexical sememes based on semantic meanings of words encoded by word embeddings. Moreover, we apply matrix factorization to learn semantic relations between sememes and words. In experiments, we take a real-world sememe knowledge base HowNet for training and evaluation, and the results reveal the effectiveness of our method for lexical sememe prediction. Our method will be of great use for annotation verification of existing noisy sememe knowledge bases and annotation suggestion of new words and phrases. Ruobing Xie, Xingchi Yuan, Zhiyuan Liu 0001, Maosong Sun 0001 |
IJCAI | 4 |
| 2017 | Fast Network Embedding Enhancement via High Order Proximity ApproximationabstractMany Network Representation Learning (NRL) methods have been proposed to learn vector representations for vertices in a network recently. In this paper, we summarize most existing NRL methods into a unified two-step framework, including proximity matrix construction and dimension reduction. We focus on the analysis of proximity matrix construction step and conclude that an NRL method can be improved by exploring higher order proximities when building the proximity matrix. We propose Network Embedding Update (NEU) algorithm which implicitly approximates higher order proximities with theoretical approximation bound and can be applied on any NRL methods to enhance their performances. We conduct experiments on multi-label classification and link prediction tasks. Experimental results show that NEU can make a consistent and significant improvement over a number of NRL methods with almost negligible running time on all three publicly available datasets. Cheng Yang 0002, Maosong Sun 0001, Zhiyuan Liu 0001, Cunchao Tu |
IJCAI | 2 |
| 2017 | Iterative Entity Alignment via Joint Knowledge EmbeddingsabstractEntity alignment aims to link entities and their counterparts among multiple knowledge graphs (KGs). Most existing methods typically rely on external information of entities such as Wikipedia links and require costly manual feature construction to complete alignment. In this paper, we present a novel approach for entity alignment via joint knowledge embeddings. Our method jointly encodes both entities and relations of various KGs into a unified low-dimensional semantic space according to a small seed set of aligned entities. During this process, we can align entities according to their semantic distance in this joint semantic space. More specifically, we present an iterative and parameter sharing method to improve alignment performance. Experiment results on real-world datasets show that, as compared to baselines, our method achieves significant improvements on entity alignment, and can further improve knowledge graph completion performance on various KGs with the favor of joint knowledge embeddings. Hao Zhu 0006, Ruobing Xie, Zhiyuan Liu 0001, Maosong Sun 0001 |
IJCAI | 4 |
| 2017 | Recent Advances on Neural Headline Generation
Ayana, Shiqi Shen, Yankai Lin 0001, Cunchao Tu, Yu Zhao 0040, Zhiyuan Liu 0001, Maosong Sun 0001 |
J. Comput. Sci. Technol. | 7 |
| 2017 | Optimizing Non-Decomposable Evaluation Metrics for Neural Machine Translation
Shiqi Shen, Yang Liu 0005, Maosong Sun 0001 |
J. Comput. Sci. Technol. | 3 |
| 2017 | Neural Parse Combination
Liner Yang, Maosong Sun 0001, Yong Cheng 0003, Zhenghao Liu 0001, Huan-Bo Luan, Yang Liu 0005 |
J. Comput. Sci. Technol. | 2 |
| 2017 | Improving Word Representations with Document LabelsabstractWord representations, aiming to build vectors for each word, have been successfully used in a variety of applications. Most word representations are learned from large amounts of documents ignoring other information. This is rather suboptimal because the side information of the documents, such as document labels, is not used in learning word representations. In this paper, we focus on how to exploit these side information to improve word representations. We propose to incorporate document labels into the learning process of word representations in two frameworks: neural network and matrix factorization. The experimental results on word analogy and word similarity task show that our models can better capture the semantic and syntactic information than the original models. Our models also improve the performance of word representations on text classification task. Liner Yang, Xinxiong Chen, Zhiyuan Liu 0001, Maosong Sun 0001 |
IEEE ACM Trans. Audio Speech Lang. Process. | 4 |
| 2017 | PRISM: Profession Identification in Social MediaabstractProfession is an important social attribute of people. It plays a crucial role in commercial services such as personalized recommendation and targeted advertising. In practice, profession information is usually unavailable due to privacy and other reasons. In this article, we explore the task of identifying user professions according to their behaviors in social media. The task confronts the following challenges that make it non-trivial: how to incorporate heterogeneous information of user behaviors, how to effectively utilize both labeled and unlabeled data, and how to exploit community structure. To address these challenges, we present a framework called Profession Identification in Social Media. It takes advantage of both personal information and community structure of users in the following aspects: (1) We present a cascaded two-level classifier with heterogeneous personal features to measure the confidence of users belonging to different professions. (2) We present a multi-training process to take advantages of both labeled and unlabeled data to enhance classification performance. (3) We design a profession identification method synthetically considering the confidences from personal features and community structure. We collect a real-world dataset to conduct experiments, and experimental results demonstrate the significant effectiveness of our method compared with other baseline methods. By applying prediction on large-scale users, we also analyze characteristics of microblog users, finding that there are significant diversities among users of different professions in demographics, social network structures, and linguistic styles. Cunchao Tu, Zhiyuan Liu 0001, Huan-Bo Luan, Maosong Sun 0001 |
ACM Trans. Intell. Syst. Technol. | 4 |
| 2017 | A Neural Network Approach to Jointly Modeling Social Networks and Mobile TrajectoriesabstractTwo characteristics of location-based services are mobile trajectories and the ability to facilitate social networking. The recording of trajectory data contributes valuable resources towards understanding users’ geographical movement behaviors. Social networking is possible when users are able to quickly connect to anyone nearby. A social network with location based services is known as location-based social network (LBSN). As shown in Cho et al. [2013], locations that are frequently visited by socially related persons tend to be correlated, which indicates the close association between social connections and trajectory behaviors of users in LBSNs. To better analyze and mine LBSN data, we need to have a comprehensive view of each of these two aspects, i.e., the mobile trajectory data and the social network. Specifically, we present a novel neural network model that can jointly model both social networks and mobile trajectories. Our model consists of two components: the construction of social networks and the generation of mobile trajectories. First we adopt a network embedding method for the construction of social networks: a networking representation can be derived for a user. The key to our model lies in generating mobile trajectories. Second, we consider four factors that influence the generation process of mobile trajectories: user visit preference, influence of friends, short-term sequential contexts, and long-term sequential contexts. To characterize the last two contexts, we employ the RNN and GRU models to capture the sequential relatedness in mobile trajectories at the short or long term levels. Finally, the two components are tied by sharing the user network representations. Experimental results on two important applications demonstrate the effectiveness of our model. In particular, the improvement over baselines is more significant when either network structure or trajectory data is sparse. Cheng Yang 0002, Maosong Sun 0001, Wayne Xin Zhao, Zhiyuan Liu 0001, Edward Y. Chang |
ACM Trans. Inf. Syst. | 2 |
| 2016 | Representation Learning of Knowledge Graphs with Entity DescriptionsabstractRepresentation learning (RL) of knowledge graphs aims to project both entities and relations into a continuous low-dimensional space. Most methods concentrate on learning representations with knowledge triples indicating relations between entities. In fact, in most knowledge graphs there are usually concise descriptions for entities, which cannot be well utilized by existing methods. In this paper, we propose a novel RL method for knowledge graphs taking advantages of entity descriptions. More specifically, we explore two encoders, including continuous bag-of-words and deep convolutional neural models to encode semantics of entity descriptions. We further learn knowledge representations with both triples and descriptions. We evaluate our method on two tasks, including knowledge graph completion and entity classification. Experimental results on real-world datasets show that, our method outperforms other baselines on the two tasks, especially under the zero-shot setting, which indicates that our method is capable of building representations for novel entities according to their descriptions. The source code of this paper can be obtained from https://github.com/xrb92/DKRL. Ruobing Xie, Zhiyuan Liu 0001, Jia Jia 0001, Huan-Bo Luan, Maosong Sun 0001 |
AAAI | 5 |
| 2016 | Building Earth Mover's Distance on Bilingual Word Embeddings for Machine TranslationabstractFollowing their monolingual counterparts, bilingual word embeddings are also on the rise. As a major application task, word translation has been relying on the nearest neighbor to connect embeddings cross-lingually. However, the nearest neighbor strategy suffers from its inherently local nature and fails to cope with variations in realistic bilingual word embeddings. Furthermore, it lacks a mechanism to deal with many-to-many mappings that often show up across languages. We introduce Earth Mover's Distance to this task by providing a natural formulation that translates words in a holistic fashion, addressing the limitations of the nearest neighbor. We further extend the formulation to a new task of identifying parallel sentences, which is useful for statistical machine translation systems, thereby expanding the application realm of bilingual word embeddings. We show encouraging performance on both tasks. Meng Zhang 0019, Yang Liu 0005, Huan-Bo Luan, Maosong Sun 0001, Tatsuya Izuha |
AAAI | 4 |
| 2016 | Semi-Supervised Learning for Neural Machine TranslationabstractWhile end-to-end neural machine translation (NMT) has made remarkable progress recently, NMT systems only rely on parallel corpora for parameter estimation. Since parallel corpora are usually limited in quantity, quality, and coverage, especially for low-resource languages, it is appealing to exploit monolingual corpora to improve NMT. We propose a semi-supervised approach for training NMT models on the concatenation of labeled (parallel corpora) and unlabeled (monolingual corpora) data. The central idea is to reconstruct the monolingual corpora using an autoencoder, in which the source-to-target and target-to-source translation models serve as the encoder and decoder, respectively. Our approach can not only exploit the monolingual corpora of the target language, but also of the source language. Experiments on the Chinese-English dataset show that our approach achieves significant improvements over state-of-the-art SMT and NMT systems. Yong Cheng 0003, Wei Xu 0005, Zhongjun He, Wei He 0014, Hua Wu 0003, Maosong Sun 0001, Yang Liu 0005 |
ACL (1) | 6 |
| 2016 | Neural Relation Extraction with Selective Attention over InstancesabstractDistant supervised relation extraction has been widely used to find novel relational facts from text.However, distant supervision inevitably accompanies with the wrong labelling problem, and these noisy data will substantially hurt the performance of relation extraction.To alleviate this issue, we propose a sentence-level attention-based model for relation extraction.In this model, we employ convolutional neural networks to embed the semantics of sentences.Afterwards, we build sentence-level attention over multiple instances, which is expected to dynamically reduce the weights of those noisy instances.Experimental results on real-world datasets show that, our model can make full use of all informative sentences and effectively reduce the influence of wrong labelled instances.Our model achieves significant and consistent improvements on relation extraction as compared with baselines.The source code of this paper can be obtained from https: //github.com/thunlp/NRE. Yankai Lin 0001, Shiqi Shen, Zhiyuan Liu 0001, Huan-Bo Luan, Maosong Sun 0001 |
ACL (1) | 5 |
| 2016 | Agreement-based Learning of Parallel Lexicons and Phrases from Non-Parallel CorporaabstractWe introduce an agreement-based approach to learning parallel lexicons and phrases from non-parallel corpora.The basic idea is to encourage two asymmetric latent-variable translation models (i.e., source-to-target and target-to-source) to agree on identifying latent phrase and word alignments.The agreement is defined at both word and phrase levels.We develop a Viterbi EM algorithm for jointly training the two unidirectional models efficiently.Experiments on the Chinese-English dataset show that agreementbased learning significantly improves both alignment and translation performance. Yang Liu 0005, Maosong Sun 0001, Huan-Bo Luan, Heng Yu 0006 |
ACL (1) | 3 |
| 2016 | Minimum Risk Training for Neural Machine TranslationabstractWe propose minimum risk training for end-to-end neural machine translation.Unlike conventional maximum likelihood estimation, minimum risk training is capable of optimizing model parameters directly with respect to arbitrary evaluation metrics, which are not necessarily differentiable.Experiments show that our approach achieves significant improvements over maximum likelihood estimation on a state-of-the-art neural machine translation system across various languages pairs.Transparent to architectures, our approach can be applied to more neural networks and potentially benefit more NLP tasks. Shiqi Shen, Yong Cheng 0003, Zhongjun He, Wei He 0014, Hua Wu 0003, Maosong Sun 0001, Yang Liu 0005 |
ACL (1) | 6 |
| 2016 | Inducing Bilingual Lexica From Non-Parallel Data With Earth Mover's Distance RegularizationabstractBeing able to induce word translations from non-parallel data is often a prerequisite for cross-lingual processing in resource-scarce languages and domains. Previous endeavors typically simplify this task by imposing the one-to-one translation assumption, which is too strong to hold for natural languages. We remove this constraint by introducing the Earth Mover’s Distance into the training of bilingual word embeddings. In this way, we take advantage of its capability to handle multiple alternative word translations in a natural form of regularization. Our approach shows significant and consistent improvements across four language pairs. We also demonstrate that our approach is particularly preferable in resource-scarce settings as it only requires a minimal seed lexicon. Meng Zhang 0019, Yang Liu 0005, Huan-Bo Luan, Yiqun Liu 0001, Maosong Sun 0001 |
COLING | 5 |
| 2016 | Neural Sentiment Classification with User and Product AttentionabstractDocument-level sentiment classification aims to predict user's overall sentiment in a document about a product.However, most of existing methods only focus on local text information and ignore the global user preference and product characteristics.Even though some works take such information into account, they usually suffer from high model complexity and only consider wordlevel preference rather than semantic levels.To address this issue, we propose a hierarchical neural network to incorporate global user and product information into sentiment classification.Our model first builds a hierarchical LSTM model to generate sentence and document representations.Afterwards, user and product information is considered via attentions over different semantic levels due to its ability of capturing crucial semantic components.The experimental results show that our model achieves significant and consistent improvements compared to all state-of-theart methods.The source code of this paper can be obtained from https://github.com/thunlp/NSC. Maosong Sun 0001, Cunchao Tu, Yankai Lin 0001, Zhiyuan Liu 0001 |
EMNLP | 2 |
| 2016 | Agreement-Based Joint Training for Bidirectional Attention-Based Neural Machine Translation
Yong Cheng 0003, Shiqi Shen, Zhongjun He, Wei He 0014, Hua Wu 0003, Maosong Sun 0001, Yang Liu 0005 |
IJCAI | 6 |
| 2016 | Knowledge Representation Learning with Entities, Attributes and Relations
Yankai Lin 0001, Zhiyuan Liu 0001, Maosong Sun 0001 |
IJCAI | 3 |
| 2016 | Max-Margin DeepWalk: Discriminative Learning of Network Representation
Cunchao Tu, Weicheng Zhang, Zhiyuan Liu 0001, Maosong Sun 0001 |
IJCAI | 4 |
| 2016 | Representation Learning of Knowledge Graphs with Hierarchical Types
Ruobing Xie, Zhiyuan Liu 0001, Maosong Sun 0001 |
IJCAI | 3 |
| 2016 | Listwise Ranking Functions for Statistical Machine TranslationabstractDecision rules play an important role in the tuning and decoding steps of statistical machine translation. The traditional decision rule selects the candidate with the greatest potential from a candidate space by examining each candidate individually. However, viewing each candidate as independent imposes a serious limitation on the translation task. We instead view the problem from a ranking perspective that naturally allows the consideration of an entire list of candidates as a whole through the adoption of a listwise ranking function. Our shift from a pointwise to a listwise perspective proves to be a simple yet powerful extension to current modeling that allows arbitrary pairwise functions to be incorporated as features, whose weights can be estimated jointly with traditional ones. We further demonstrate that our formulation encompasses the minimum Bayes risk (MBR) approach, another decision rule that considers restricted listwise information, as a special case. Experiments show that our approach consistently outperforms the baseline and MBR methods across the considered test sets. Meng Zhang 0019, Yang Liu 0005, Huan-Bo Luan, Maosong Sun 0001 |
IEEE ACM Trans. Audio Speech Lang. Process. | 4 |
| 2015 | Learning Entity and Relation Embeddings for Knowledge Graph CompletionabstractKnowledge graph completion aims to perform link prediction between entities. In this paper, we consider the approach of knowledge graph embeddings. Recently, models such as TransE and TransH build entity and relation embeddings by regarding a relation as translation from head entity to tail entity. We note that these models simply put both entities and relations within the same semantic space. In fact, an entity may have multiple aspects and various relations may focus on different aspects of entities, which makes a common space insufficient for modeling. In this paper, we propose TransR to build entity and relation embeddings in separate entity space and relation spaces. Afterwards, we learn embeddings by first projecting entities from entity space to corresponding relation space and then building translations between projected entities. In experiments, we evaluate our models on three tasks including link prediction, triple classification and relational fact extraction. Experimental results show significant and consistent improvements compared to state-of-the-art baselines including TransE and TransH. Yankai Lin 0001, Zhiyuan Liu 0001, Maosong Sun 0001, Yang Liu 0107, Xuan Zhu 0006 |
AAAI | 3 |
| 2015 | Topical Word EmbeddingsabstractMost word embedding models typically represent each word using a single vector, which makes these models indiscriminative for ubiquitous homonymy and polysemy. In order to enhance discriminativeness, we employ latent topic models to assign topics for each word in the text corpus, and learn topical word embeddings (TWE) based on both words and their topics. In this way, contextual word embeddings can be flexibly obtained to measure contextual word similarity. We can also build document representations, which are more expressive than some widely-used document models such as latent topic models. In the experiments, we evaluate the TWE models on two tasks, contextual word similarity and text classification. The experimental results show that our models outperform typical word embedding models including the multi-prototype version on contextual word similarity, and also exceed latent topic models and other representative document models on text classification. Yang Liu 0005, Zhiyuan Liu 0001, Tat-Seng Chua, Maosong Sun 0001 |
AAAI | 4 |
| 2015 | Contrastive Unsupervised Word Alignment with Non-Local FeaturesabstractWord alignment is an important natural language processing task that indicates the correspondence between natural languages. Recently, unsupervised learning of log-linear models for word alignment has received considerable attention as it combines the merits of generative and discriminative approaches. However, a major challenge still remains: it is intractable to calculate the expectations of non-local features that are critical for capturing the divergence between natural languages. We propose a contrastive approach that aims to differentiate observed training examples from noises. It not only introduces prior knowledge to guide unsupervised learning but also cancels out partition functions. Based on the observation that the probability mass of log-linear models for word alignment is usually highly concentrated, we propose to use top-$n$ alignments to approximate the expectations with respect to posterior distributions. This allows for efficient and accurate calculation of expectations of non-local features. Experiments show that our approach achieves significant improvements over state-of-the-art unsupervised word alignment methods. Yang Liu 0005, Maosong Sun 0001 |
AAAI | 2 |
| 2015 | Phrase Type Sensitive Tensor Indexing Model for Semantic CompositionabstractCompositional semantic aims at constructing the meaning of phrases or sentences according to the compositionality of word meanings. In this paper, we propose to synchronously learn the representations of individual words and extracted high-frequency phrases. Representations of extracted phrases are considered as gold standard for constructing more general operations to compose the representation of unseen phrases. We propose a grammatical type specific model that improves the composition flexibility by adopting vector-tensor-vector operations. Our model embodies the compositional characteristics of traditional additive and multiplicative model. Empirical result shows that our model outperforms state-of-the-art composition methods in the task of computing phrase similarities. Yu Zhao 0040, Zhiyuan Liu 0001, Maosong Sun 0001 |
AAAI | 3 |
| 2015 | Modeling Relation Paths for Representation Learning of Knowledge BasesabstractRepresentation learning of knowledge bases aims to embed both entities and relations into a low-dimensional space.Most existing methods only consider direct relations in representation learning.We argue that multiple-step relation paths also contain rich inference patterns between entities, and propose a path-based representation learning model.This model considers relation paths as translations between entities for representation learning, and addresses two key challenges: (1) Since not all relation paths are reliable, we design a path-constraint resource allocation algorithm to measure the reliability of relation paths.(2) We represent relation paths via semantic composition of relation embeddings.Experimental results on real-world datasets show that, as compared with baselines, our model achieves significant and consistent improvements on knowledge base completion and relation extraction from text.The source code of this paper can be obtained from https://github.com/mrlyk423/relation_extraction. Yankai Lin 0001, Zhiyuan Liu 0001, Huan-Bo Luan, Maosong Sun 0001, Siwei Rao |
EMNLP | 4 |
| 2015 | Generalized Agreement for Bidirectional Word AlignmentabstractWhile agreement-based joint training has proven to deliver state-of-the-art alignment accuracy, the produced word alignments are usually restricted to one-toone mappings because of the hard constraint on agreement.We propose a general framework to allow for arbitrary loss functions that measure the disagreement between asymmetric alignments.The loss functions can not only be defined between asymmetric alignments but also between alignments and other latent structures such as phrase segmentations.We use a Viterbi EM algorithm to train the joint model since the inference is intractable.Experiments on Chinese-English translation show that joint training with generalized agreement achieves significant improvements over two state-ofthe-art alignment methods. Yang Liu 0005, Maosong Sun 0001, Huan-Bo Luan, Heng Yu 0006 |
EMNLP | 3 |
| 2015 | Online Learning of Interpretable Word EmbeddingsabstractWord embeddings encode semantic meanings of words into low-dimension word vectors.In most word embeddings, one cannot interpret the meanings of specific dimensions of those word vectors.Nonnegative matrix factorization (NMF) has been proposed to learn interpretable word embeddings via non-negative constraints.However, NMF methods suffer from scale and memory issue because they have to maintain a global matrix for learning.To alleviate this challenge, we propose online learning of interpretable word embeddings from streaming text data.Experiments show that our model consistently outperforms the state-of-the-art word embedding methods in both representation ability and interpretability.The source code of this paper can be obtained from http: //github.com/skTim/OIWE. Hongyin Luo, Zhiyuan Liu 0001, Huan-Bo Luan, Maosong Sun 0001 |
EMNLP | 4 |
| 2015 | Consistency-Aware Search for Word AlignmentabstractAs conventional word alignment search algorithms usually ignore the consistency constraint in translation rule extraction, improving alignment accuracy does not necessarily increase translation quality.We propose to use coverage, which reflects how well extracted phrases can recover the training data, to enable word alignment to model consistency and correlate better with machine translation.This can be done by introducing an objective that maximizes both alignment model score and coverage.We introduce an efficient algorithm to calculate coverage on the fly during search.Experiments show that our consistency-aware search algorithm significantly outperforms both generative and discriminative alignment approaches across various languages and translation models. Shiqi Shen, Yang Liu 0005, Maosong Sun 0001, Huan-Bo Luan |
EMNLP | 3 |
| 2015 | Joint Learning of Character and Word Embeddings
Xinxiong Chen, Lei Xu 0040, Zhiyuan Liu 0001, Maosong Sun 0001, Huan-Bo Luan |
IJCAI | 4 |
| 2015 | Iterative Learning of Parallel Lexicons and Phrases from Non-Parallel Corpora
Meiping Dong, Yang Liu 0005, Huan-Bo Luan, Maosong Sun 0001, Tatsuya Izuha, Dakun Zhang |
IJCAI | 4 |
| 2015 | Network Representation Learning with Rich Text Information
Cheng Yang 0002, Zhiyuan Liu 0001, Deli Zhao, Maosong Sun 0001, Edward Y. Chang |
IJCAI | 4 |
| 2015 | Representation Learning for Measuring Entity Relatedness with Rich Information
Yu Zhao 0040, Zhiyuan Liu 0001, Maosong Sun 0001 |
IJCAI | 3 |
| 2015 | Estimating translation probabilities for social tag suggestionabstractThe task of social tag suggestion is to recommend tags automatically for a user when he or she wants to annotate an online resource. In this study, we focus on how to make use of the text description of a resource to suggest tags. It is intuitive to select significant words from the text description of a source as the suggested tags. However, since users can arbitrarily annotate any tags to a resource, tag suggestion suffers from the vocabulary gap issue — the appropriate tags of a resource may be statistically insignificant or even do not appear in the corresponding description. In order to solve the vocabulary gap issue, in this paper we present a new perspective on social tag suggestion. By considering both a description and tags as summaries of a given resource composed in two languages, tag suggestion can be regarded as a translation from description to tags. We propose two methods to estimate the translation probabilities between words in descriptions and tags. Based on the translation probabilities between words and tags estimated for a large collection of description-tags pairs, we can suggest tags according to the words in a resource description. Experiments on real-world datasets indicate that our methods outperform other methods in precision, recall and F-measure. Moreover, our methods are relatively simple and efficient, which makes them practical for Web applications. Xinxiong Chen, Zhiyuan Liu 0001, Maosong Sun 0001 |
Expert Syst. Appl. | 3 |
| 2015 | Tag Correspondence Model for User Tag Suggestion
Cunchao Tu, Zhiyuan Liu 0001, Maosong Sun 0001 |
J. Comput. Sci. Technol. | 3 |
| 2014 | ImgWordle: image and text visualization for events in microblogging servicesabstractWith the wide usage of microblogging services, microposts grow at high rate and provide a rich source of information related to important social events and trends. However, analyzing microposts is challenging due to its high complexity and large volume. In this paper, we present ImgWordle, an interactive visualization prototype to help people perceive and analyze the image and text information in microblogging services. The prototype extends tag cloud by involving images as well as words, and provides multiple coordinated views for a given event, including keywords of various topics, representative images, geographic sentiments and amounts of microposts, and popular microposts over time. We implement ImgWordle on Sina Weibo, the most popular microblogging service in China, and illustrate the usefulness of this visual interface. Chong Kuang, Jiayu Tang, Zhiyuan Liu 0001, Maosong Sun 0001 |
AVI | 4 |
| 2014 | Query Lattice for Translation Retrieval
Meiping Dong, Yong Cheng 0003, Yang Liu 0005, Jia Xu 0004, Maosong Sun 0001, Tatsuya Izuha |
COLING | 5 |
| 2014 | A Neural Reordering Model for Phrase-based Translation
Peng Li 0030, Yang Liu 0005, Maosong Sun 0001, Tatsuya Izuha, Dakun Zhang |
COLING | 3 |
| 2014 | A Unified Model for Word Sense Representation and DisambiguationabstractMost word representation methods assume that each word owns a single semantic vec-tor. This is usually problematic because lexical ambiguity is ubiquitous, which is also the problem to be resolved by word sense disambiguation. In this paper, we present a unified model for joint word sense representation and disambiguation, which will assign distinct representation-s for each word sense.1 The basic idea is that both word sense representation (WS-R) and word sense disambiguation (WS-D) will benefit from each other: (1) high-quality WSR will capture rich informa-tion about words and senses, which should be helpful for WSD, and (2) high-quality WSD will provide reliable disambiguat-ed corpora for learning better sense rep-resentations. Experimental results show that, our model improves the performance of contextual word similarity compared to existing WSR methods, outperforms state-of-the-art supervised methods on domain-specific WSD, and achieves competitive performance on coarse-grained all-words WSD. 1 Xinxiong Chen, Zhiyuan Liu 0001, Maosong Sun 0001 |
EMNLP | 3 |
| 2013 | An Extended GHKM Algorithm for Inducing Lambda-SCFGabstractSemantic parsing, which aims at mapping a natural language (NL) sentence into its formal meaning representation (e.g., logical form), has received increasing attention in recent years. While synchronous context-free grammar (SCFG) augmented with lambda calculus (lambda-SCFG) provides an effective mechanism for semantic parsing, how to learn such lambda-SCFG rules still remains a challenge because of the difficulty in determining the correspondence between NL sentences and logical forms. To alleviate this structural divergence problem, we extend the GHKM algorithm, which is a state-of-the-art algorithm for learning synchronous grammars in statistical machine translation, to induce lambda-SCFG from pairs of NL sentences and logical forms. By treating logical forms as trees, we reformulate the theory behind GHKM that gives formal semantics to the alignment between NL words and logical form tokens. Experiments on the GEOQUERY dataset show that our semantic parser achieves an F-measure of 90.2%, the best result published to date. Peng Li 0030, Yang Liu 0005, Maosong Sun 0001 |
AAAI | 3 |
| 2013 | Recursive Autoencoders for ITG-Based TranslationabstractWhile inversion transduction grammar (ITG) is well suited for modeling ordering shifts between languages, how to make applying the two reordering rules (i.e., straight and inverted) dependent on actual blocks being merged remains a challenge.Unlike previous work that only uses boundary words, we propose to use recursive autoencoders to make full use of the entire merging blocks alternatively.The recursive autoencoders are capable of generating vector space representations for variable-sized phrases, which enable predicting orders to exploit syntactic and semantic information from a neural language modeling's perspective.Experiments on the NIST 2008 dataset show that our system significantly improves over the MaxEnt classifier by 1.07 BLEU points. Peng Li 0030, Yang Liu 0005, Maosong Sun 0001 |
EMNLP | 3 |
| 2013 | HMeanMax: Placing HMAX and HoG into a unified frameworkabstractRecently, the bio-inspired model HMAX has attracted much attention for its amazing biological inspired structure and comparable performance to the state-of-the-art computer vision algorithms. The success of HMAX leads to our further exploration on it especially the connection between the biological mechanism and the engineering validity. By detailedly analyzing HMAX and a totally engineering-driven approach HoG, we find such two methods have similar structures excepts the different pooling strategies, max versus mean, thus can be placed into a unified framework. Therefore, we present a unified framework named HMeanMax to integrate HMAX and HoG via combining multiple types of pooling into a single hierarchical feature extractor. All the experimental results support our findings. Yan Zhang 0031, Qixia Jiang, Maosong Sun 0001 |
IJCNN | 3 |
| 2013 | Measuring and Visualizing Interest Similarity between Microblog Users
Jiayu Tang, Zhiyuan Liu 0001, Maosong Sun 0001 |
WAIM | 3 |
| 2012 | Topical Word Trigger Model for Keyphrase Extraction
Zhiyuan Liu 0001, Maosong Sun 0001 |
COLING | 3 |
| 2012 | Monte Carlo Methods for Maximum Margin Supervised Topic ModelsabstractAn effective strategy to exploit the supervising side information for discovering predictive topic representations is to impose discriminative constraints induced by such information on the posterior distributions under a topic model. This strategy has been adopted by a number of supervised topic models, such as MedLDA, which employs max-margin posterior constraints. However, unlike the likelihood-based supervised topic models, of which posterior inference can be carried out using the Bayes' rule, the max-margin posterior constraints have made Monte Carlo methods infeasible or at least not directly applicable, thereby limited the choice of inference algorithms to be based on variational approximation with strict mean field assumptions. In this paper, we develop two efficient Monte Carlo methods under much weaker assumptions for max-margin supervised topic models based on an importance sampler and a collapsed Gibbs sampler, respectively, in a convex dual formulation. We report thorough experimental results that compare our approach favorably against existing alternatives in both accuracy and efficiency. Qixia Jiang, Jun Zhu 0001, Maosong Sun 0001, Eric P. Xing |
NIPS | 3 |
| 2012 | Mining the interests of Chinese microbloggers via keyword extraction
Zhiyuan Liu 0001, Xinxiong Chen, Maosong Sun 0001 |
Frontiers Comput. Sci. China | 3 |
| 2011 | Fast Query Recommendation by SearchabstractQuery recommendation can not only effectively facilitate users to obtain their desired information but alsoincrease ads’ click-through rates. This paper presentsa general and highly efficient method for query recommendation. Given query sessions, we automatically generate many similar and dissimilar query-pairs as the prior knowledge. Then we learn a transformation from the prior knowledge to move similar queries closer such that similar queries tend to have similar hash values.This is formulated as minimizing the empirical error on the prior knowledge while maximizing the gap between the data and some partition hyperplanes randomly generated in advance. In the recommendation stage, we search queries that have similar hash values to the given query, rank the found queries and return the top K queries as the recommendation result. All the experimental results demonstrate that our method achieves encouraging results in terms of efficiency and recommendation performance. Qixia Jiang, Maosong Sun 0001 |
AAAI | 2 |
| 2011 | Semi-Supervised SimHash for Efficient Document Similarity Search
Qixia Jiang, Maosong Sun 0001 |
ACL | 2 |
| 2011 | Automatic Keyphrase Extraction by Bridging Vocabulary Gap
Zhiyuan Liu 0001, Xinxiong Chen, Yabin Zheng, Maosong Sun 0001 |
CoNLL | 4 |
| 2011 | A Simple Word Trigger Method for Social Tag Suggestion
Zhiyuan Liu 0001, Xinxiong Chen, Maosong Sun 0001 |
EMNLP | 3 |
| 2011 | CHIME: An Efficient Error-Tolerant Chinese Pinyin Input Method
Yabin Zheng, Chen Li 0001, Maosong Sun 0001 |
IJCAI | 3 |
| 2011 | Extract Chinese Unknown Words from a Large-scale Corpus Using Morphological and Distributional Evidences
Kaixu Zhang, Ruining Wang, Ping Xue 0002, Maosong Sun 0001 |
IJCNLP | 4 |
| 2011 | User Behaviors in Related Word Retrieval and New Word Detection: A Collaborative PerspectiveabstractNowadays, user behavior analysis and collaborative filtering have drawn a large body of research in the machine learning community. The goal is either to enhance the user experience or discover useful information hidden in the data. In this article, we conduct extensive experiments on a Chinese input method data set, which keeps the word lists that users have used. Then, from the collaborative perspective, we aim to solve two tasks in natural language processing, that is, related word retrieval and new word detection. Motivated by the observation that two words are usually highly related to each other if they co-occur frequently in users’ records, we propose a novel semantic relatedness measure between words that takes both user behaviors and collaborative filtering into consideration. We utilize this measure to perform related word retrieval and new word detection tasks. Experimental results on both tasks indicate the applicability and effectiveness of our method. Zhiyuan Liu 0001, Yabin Zheng, Lixing Xie, Maosong Sun 0001, Liyun Ru, Yang Zhang 0094 |
ACM Trans. Asian Lang. Inf. Process. | 4 |
| 2011 | PLDA+: Parallel latent dirichlet allocation with data placement and pipeline processingabstractPrevious methods of distributed Gibbs sampling for LDA run into either memory or communication bottlenecks. To improve scalability, we propose four strategies: data placement , pipeline processing , word bundling , and priority-based scheduling . Experiments show that our strategies significantly reduce the unparallelizable communication bottleneck and achieve good load balancing, and hence improve scalability of LDA. Zhiyuan Liu 0001, Edward Y. Chang, Maosong Sun 0001 |
ACM Trans. Intell. Syst. Technol. | 4 |
| 2010 | Chinese New Word Detection from Query Logs
Yan Zhang 0031, Maosong Sun 0001, Yang Zhang 0094 |
ADMA (2) | 2 |
| 2010 | Explore the Structure of Social Tags by Subsumption Relations
Xiance Si, Zhiyuan Liu 0001, Maosong Sun 0001 |
COLING | 3 |
| 2010 | Automatic Keyphrase Extraction via Topic Decomposition
Zhiyuan Liu 0001, Wenyi Huang, Yabin Zheng, Maosong Sun 0001 |
EMNLP | 4 |
| 2010 | Tag Allocation Model: Model Noisy Social Annotations by Reason FindingabstractWe propose the Tag Allocation Model (TAM) to model social annotation data. TAM is a probabilistic generative model, its key feature is finding the latent reason for each tag. A latent reason can be any discrete features of the document (such as words) or a global noise variable. Inferring the reason for each tag helps TAM reduce the ambiguity of a document with multiple tags. By introducing noise as a reason, TAM can handle noise tags naturally. We perform experiments on three real world data sets. The results show that TAM outperforms state-of-the-art approaches in both held-out perplexity and tag recommendation accuracy. Xiance Si, Maosong Sun 0001 |
Web Intelligence | 2 |
| 2010 | Confucius and Its Intelligent Disciples: Integrating Social with SearchabstractQ&A sites continue to flourish as a large number of users rely on them as useful substitutes for incomplete or missing search results. In this paper, we present our experience with developing Confucius, a Google Q&A service launched in 21 countries and four languages by the end of 2009. Confucius employs six data mining subroutines to harness synergy between web search and social networks. We present these subroutines' design goals, algorithms, and their effects on service quality. We also describe techniques for and experience with scaling the subroutines to mine massive data sets. Xiance Si, Edward Y. Chang, Zoltán Gyöngyi, Maosong Sun 0001 |
Proc. VLDB Endow. | 4 |
| 2009 | Community Detection on Weighted Networks: A Variational Bayesian Method
Qixia Jiang, Yan Zhang 0031, Maosong Sun 0001 |
ACML | 3 |
| 2009 | Clustering to Find Exemplar Terms for Keyphrase Extraction
Zhiyuan Liu 0001, Peng Li 0030, Yabin Zheng, Maosong Sun 0001 |
EMNLP | 4 |
| 2009 | Incorporating User Behaviors in New Word Detection
Yabin Zheng, Zhiyuan Liu 0001, Maosong Sun 0001, Liyun Ru, Yang Zhang 0094 |
IJCAI | 3 |
| 2009 | Incorporate Web Search Technology to Solve Out-of-Vocabulary Words in Chinese Word Segmentation
Maosong Sun 0001 |
PACLIC | 2 |
| 2009 | Quantifying Asymmetric Semantic Relations from Query Logs by Resource Allocation
Zhiyuan Liu 0001, Yabin Zheng, Maosong Sun 0001 |
PAKDD | 3 |
| 2009 | Punctuation as Implicit Annotations for Chinese Word SegmentationabstractWe present a Chinese word segmentation model learned from punctuation marks which are perfect word delimiters. The learning is aided by a manually segmented corpus. Our method is considerably more effective than previous methods in unknown word recognition. This is a step toward addressing one of the toughest problems in Chinese word segmentation. Zhongguo Li, Maosong Sun 0001 |
Comput. Linguistics | 2 |
| 2008 | Full-reference quality diagnosis for video summaryabstractAs video summarization techniques have attracted more and more attention for efficient multimedia data management, objective quality assessment of video summary is desired. To address the lack of automatic evaluation techniques, this paper proposes a 3C-diagnosis algorithm to diagnose the video summary from the perspective of coverage, conciseness, and coherence. The candidate summary is first aligned against the reference summary. Then the coverage of the candidate summary is calculated according to the information bearing of the matcThing frames and the information loss of the missing frames. The conciseness is calculated based on the unwanted information contained in the candidate summary, and the coherence is calculated based on the ratio of the appearances of the frame loss for the aligned candidate summary. The proposed techniques are experimented on a standard dataset of TRECVID 2007 and show good performance. Yan Liu 0004, Yan Zhang 0031, Maosong Sun 0001, Wenjie Li 0002 |
ICME | 3 |
| 2008 | Asymmetrical query recommendation method based on bipartite network resource allocationabstractThis paper presents a new query recommendation method that generates recommended query list by mining large-scale user logs. Starting from the user logs of click-through data, we construct a bipartite network where the nodes on one side correspond to unique queries, on the other side to unique URLs. Inspired by the bipartite network based resource allocation method, we try to extract the hidden information from the Query-URL bipartite network. The recommended queries generated by the method are asymmetrical which means two related queries may have different strength to recommend each other. To evaluate the method, we use one week user logs from Chinese search engine Sogou. The method is not only `content ignorant', but also can be easily implemented in a paralleled manner, which is feasible for commercial search engines to handle large scale user logs. Zhiyuan Liu 0001, Maosong Sun 0001 |
WWW | 2 |
| 2007 | Exploiting Category Information and Document Information to Improve Term Weighting for Text Categorization
Maosong Sun 0001 |
CICLing | 2 |
| 2007 | Scalable Term Selection for Text Categorization
Maosong Sun 0001 |
EMNLP-CoNLL | 2 |
| 2006 | A Comparison and Semi-Quantitative Analysis of Words and Character-Bigrams as Features in Chinese Text CategorizationabstractWords and character-bigrams are both used as features in Chinese text processing tasks, but no systematic comparison or analysis of their values as features for Chinese text categorization has been reported heretofore. We carry out here a full performance comparison between them by experiments on various document collections (including a manually word-segmented corpus as a golden standard), and a semi-quantitative analysis to elucidate the characteristics of their behavior; and try to provide some preliminary clue for feature term choice (in most cases, character-bigrams are better than words) and dimensionality setting in text categorization systems. Maosong Sun 0001 |
ACL | 2 |
| 2006 | Automatic Image Annotation Based on WordNet and Hierarchical Ensembles
Wei Li 0026, Maosong Sun 0001 |
CICLing | 2 |
| 2006 | Word Frequency Approximation for Chinese Without Using Manually-Annotated Corpus
Maosong Sun 0001, Zhengcao Zhang, Benjamin Ka-Yin T'sou, Huaming Lu |
CICLing | 1 |
| 2006 | Knowledge representation and reasoning based on entity and relation propagation diagram/tree
Xinghua Fan, Maosong Sun 0001 |
Intell. Data Anal. | 2 |
| 2005 | Merging Case Relations into VSM to Improve Information Retrieval Precision
Maosong Sun 0001, Shaoming Liu |
CICLing | 2 |
| 2005 | Word Extraction Based on Semantic Constraints in Chinese Word-Formation
Maosong Sun 0001, Shengfen Luo, Benjamin Ka-Yin T'sou |
CICLing | 1 |
| 2005 | A Method of Recognizing Entity and Relation
Xinghua Fan, Maosong Sun 0001 |
IJCNLP | 2 |
| 2005 | Classifying Chinese Texts in Two Steps
Xinghua Fan, Maosong Sun 0001, Key-Sun Choi |
IJCNLP | 2 |
| 2005 | Automatic Image Annotation Using Maximum Entropy Model
Wei Li 0026, Maosong Sun 0001 |
IJCNLP | 2 |
| 2004 | Raising High-Degree Overlapped Character Bigrams into Trigrams for Dimensionality Reduction in Chinese Text Categorization
Xue Dejun, Maosong Sun 0001 |
CICLing | 2 |
| 2004 | Eliminating High-Degree Biased Character Bigrams for Dimensionality Reduction in Chinese Text Categorization
Xue Dejun, Maosong Sun 0001 |
ECIR | 2 |
| 2003 | A Study on Feature Weighting in Chinese Text Categorization
Xue Dejun, Maosong Sun 0001 |
CICLing | 2 |
| 2003 | Chinese Text Categorization Based on the Binary Weighting Model with Non-binary Smoothing
Xue Dejun, Maosong Sun 0001 |
ECIR | 2 |
| 2002 | Covering Ambiguity Resolution in Chinese Word Segmentation Based on Contextual Information
Maosong Sun 0001, Benjamin Ka-Yin T'sou |
COLING | 2 |
| 2002 | Identification of Chinese Personal Names in Unrestricted Texts
Lawrence Y. L. Cheung, Benjamin Ka-Yin T'sou, Maosong Sun 0001 |
PACLIC | 3 |
| 2001 | A prototype of Chinese search engine based on word segmentation techniquesabstractThe aim of this paper is to reveal and stress the role of word segmentation in implementing Chinese search engines on the WWW. CSeg and Tag1.1, an integrated system of word segmentation and part-of-speech tagging for unrestricted Chinese texts, along with some basic ideas of its design, are then presented. A prototype of the Chinese search engine WalkerSun, which makes use of CSeg and Tag1.1 as its underlying NLP tool, is introduced briefly. Maosong Sun 0001 |
SMC | 2 |
| 2000 | Hua Yu: A Word-segmented and Part-Of-Speech Tagged Chinese Corpus
Maosong Sun 0001, Honglin Sun, Changning Huang, Zhang Pu, Xing Hongbing, Zhou Qiang |
LREC | 1 |
| 1995 | Ambiguity Resolution in Chinese Word Segmentation
Maosong Sun 0001, Benjamin Ka-Yin T'sou |
PACLIC | 1 |