← Search

Wenge Rong

21 accepted papers

2025

CogAtom: From Cognitive Atoms to Olympiad-level Mathematical Reasoning in Large Language Models

EMNLP 2025

Mathematical reasoning poses significant challenges for Large Language Models (LLMs) due to its demand for multi-step reasoning and abstract conceptual integration. While recent test-time scaling techniques rely heavily on high-quality, challenging problems, the scarcity of Olympiad-level math probl

2025

Disentangling Preference Representation and Text Generation for Efficient Individual Preference Alignment

COLING 2025main

Aligning Large Language Models (LLMs) with general human preferences has been proved crucial in improving the interaction quality between LLMs and human. However, human values are inherently diverse among different individuals, making it insufficient to align LLMs solely with general preferences. To…

2025

Enhancing LLMs via High-Knowledge Data Selection

AAAI 2025technical

The performance of Large Language Models (LLMs) is intrinsically linked to the quality of its training data. Although several studies have proposed methods for high-quality data selection, they do not consider the importance of knowledge richness in text corpora. In this paper, we propose a novel an…

Cited by 0SourcePDFScholar
2025

PIC: Unlocking Long-Form Text Generation Capabilities of Large Language Models via Position ID Compression

ACL 2025long

Long-context understanding is crucial for large language models (LLMs) and has become a fundamental capability for most LLMs. However, beyond the focus on “input-long”, the ability to “output-long” is equally significant, yet it remains underexplored. To address this limitation, we propose a simple,…

Cited by 0SourcePDFScholar
2025

Selecting Demonstrations for Many-Shot In-Context Learning via Gradient Matching

ACL 2025finding

In-Context Learning (ICL) empowers Large Language Models (LLMs) for rapid task adaptation without Fine-Tuning (FT), but its reliance on demonstration selection remains a critical challenge. While many-shot ICL shows promising performance through scaled demonstrations, the selection method for many-s…

2024

Leveraging Estimated Transferability Over Human Intuition for Model Selection in Text Ranking

EMNLP 2024main

Text ranking has witnessed significant advancements, attributed to the utilization of dual-encoder enhanced by Pre-trained Language Models (PLMs). Given the proliferation of available PLMs, selecting the most effective one for a given dataset has become a non-trivial challenge. As a promising altern…

2024

ProCQA: A Large-scale Community-based Programming Question Answering Dataset for Code Search

COLING 2024main

Retrieval-based code question answering seeks to match user queries in natural language to relevant code snippets. Previous approaches typically rely on pretraining models using crafted bi-modal and uni-modal datasets to align text and code representations. In this paper, we introduce ProCQA, a larg…

2024

RoleAgent: Building, Interacting, and Benchmarking High-quality Role-Playing Agents from Scripts

NeurIPS 2024poster

Believable agents can empower interactive applications ranging from immersive environments to rehearsal spaces for interpersonal communication. Recently, generative agents have been proposed to simulate believable human behavior by using Large Language Models. However, the existing method heavily re…

Cited by 1SourcePDFScholar
2023

Discovering Sounding Objects by Audio Queries for Audio Visual Segmentation

IJCAI 2023poster

Audio visual segmentation (AVS) aims to segment the sounding objects for each frame of a given video. To distinguish the sounding objects from silent ones, both audio-visual semantic correspondence and temporal interaction are required. The previous method applies multi-frame cross-modal attention t…

Cited by 18SourcePDFScholar
2023

Evaluating Open-Domain Dialogues in Latent Space with Next Sentence Prediction and Mutual Information

ACL 2023long

The long-standing one-to-many issue of the open-domain dialogues poses significant challenges for automatic evaluation methods, i.e., there may be multiple suitable responses which differ in semantics for a given conversational context. To tackle this challenge, we propose a novel learning-based aut…

2023

How to Determine the Most Powerful Pre-trained Language Model without Brute Force Fine-tuning? An Empirical Survey

EMNLP 2023long findings

Transferability estimation has been attached to great attention in the computer vision fields. Researchers try to estimate with low computational cost the performance of a model when transferred from a source task to a given target task. Considering the effectiveness of such estimations, the commun…

Cited by 0SourcecodeScholar
2023

Permutation Invariant Training for Paraphrase Identification

ICASSP 2023accepted

Identifying sentences sharing similar meanings is crucial to speech and text understandings. Although currently popular cross-encoder solutions with pre-trained language models as backbone have achieved remarkable performance, they suffer from the lack of the permutation invariance or symmetry that…

Cited by 0SourceScholar
2023

Token-Level Self-Evolution Training for Sequence-to-Sequence Learning

ACL 2023short

Adaptive training approaches, widely used in sequence-to-sequence models, commonly reweigh the losses of different target tokens based on priors, e.g. word frequency. However, most of them do not consider the variation of learning difficulty in different training steps, and overly emphasize the lear…

Cited by 23SourcePDFScholar
2023

Transformer-Patcher: One Mistake Worth One Neuron

ICLR 2023poster

Large Transformer-based Pretrained Language Models (PLMs) dominate almost all Natural Language Processing (NLP) tasks. Nevertheless, they still make mistakes from time to time. For a model deployed in an industrial environment, fixing these mistakes quickly and robustly is vital to improve user expe…

2022

Improving Variational Autoencoders with Density Gap-based Regularization

NeurIPS 2022accept

Variational autoencoders (VAEs) are one of the most powerful unsupervised learning frameworks in NLP for latent representation learning and latent-directed generation. The classic optimization goal of VAEs is to maximize the Evidence Lower Bound (ELBo), which consists of a conditional likelihood for…

2022

Mixture of Attention Heads: Selecting Attention Heads Per Token

EMNLP 2022main

Mixture-of-Experts (MoE) networks have been proposed as an efficient way to scale up model capacity and implement conditional computing. However, the study of MoE components mostly focused on the feedforward layer in Transformer architecture. This paper proposes the Mixture of Attention Heads (MoA),…

2021

Enhancing Dual-Encoders with Question and Answer Cross-Embeddings for Answer Retrieval

EMNLP 2021finding

Dual-Encoders is a promising mechanism for answer retrieval in question answering (QA) systems. Currently most conventional Dual-Encoders learn the semantic representations of questions and answers merely through matching score. Researchers proposed to introduce the QA interaction features in scorin…

2021

Improving Dialogue Response Generation Via Knowledge Graph Filter

ICASSP 2021accepted

Current generative dialogue systems tend to produce generic dialog responses, which lack useful information and semantic coherence. An promising method to alleviate this problem is to integrate knowledge triples from knowledge base. However, current approaches mainly augment Seq2Seq framework with k…

Cited by 0SourceScholar
2021

Paragraph Level Multi-Perspective Context Modeling for Question Generation

ICASSP 2021accepted

Proper understanding of paragraph is essential for question generation task since the semantic interaction is complicated among sentences. How to integrate long text paragraph information into question generation is still a challenge. In this research, we proposed a multi-perspective paragraph conte…

Cited by 0SourceScholar
2021

Topic-Aware Dialogue Generation with Two-Hop Based Graph Attention

ICASSP 2021accepted

Generating on-topic responses and understanding the background information of context are both significant for dialogue generation. However, few works simultaneously concentrate on these two issues. For this purpose, we propose an open-domain topic-aware dialogue generation model via joint learning.…

Cited by 0SourceScholar
2019

Deep Hybrid Networks Based Response Selection for Multi-turn Dialogue Systems

ICASSP 2019accepted

Proper response selection is an important challenge for a meaningful multi-turn dialogue. To this end, not only the coherence among the whole dialogue but also the interaction between utterance in adjacent turns need to be properly employed as the context for response selection. In this paper, we pr…

Cited by 0SourceScholar