← Search

Jianshu Chen

31 accepted papers

2026

CoMem: Context Management with A Decoupled Long-Context Model

ICML 2026poster

Context management enables agentic models to solve long-horizon tasks through iterative summarization of previous interaction histories. However, this process typically incurs substantial decoding overhead for the extra summarization tokens, which significantly affect the end-to-end response latency…

Cited by 0SourceScholar
2024

Fact-and-Reflection (FaR) Improves Confidence Calibration of Large Language Models

ACL 2024findings

For a LLM to be trustworthy, its confidence level should be well-calibrated with its actual performance. While it is now common sense that LLM performances are greatly impacted by prompts, the confidence calibration in prompting LLMs has yet to be thoroughly explored.In this paper, we explore how di…

2024

From Language Modeling to Instruction Following: Understanding the Behavior Shift in LLMs after Instruction Tuning

NAACL 2024long

Large Language Models (LLMs) have achieved remarkable success, where instruction tuning is the critical step in aligning LLMs with user intentions. In this work, we investigate how the instruction tuning adjusts pre-trained models with a focus on intrinsic changes. Specifically, we first develop sev…

2024

MMC: Advancing Multimodal Chart Understanding with Large-scale Instruction Tuning

NAACL 2024long

With the rapid development of large language models (LLMs) and their integration into large multimodal models (LMMs), there has beenimpressive progress in zero-shot completion of user-oriented vision-language tasks. However, a gap remains in the domain of chartimage understanding due to the distinct…

2024

Rewards-in-Context: Multi-objective Alignment of Foundation Models with Dynamic Preference Adjustment

ICML 2024poster

We consider the problem of multi-objective alignment of foundation models with human preferences, which is a critical step towards helpful and harmless AI systems. However, it is generally costly and unstable to fine-tune large foundation models using reinforcement learning (RL), and the multi-dimen…

2024

Skills-in-Context: Unlocking Compositionality in Large Language Models

EMNLP 2024finding

We investigate how to elicit compositional generalization capabilities in large language models (LLMs). Compositional generalization empowers LLMs to solve complex problems by combining foundational skills, a critical reasoning ability akin to human intelligence. However, even the most advanced LLMs…

Cited by 1SourcePDFScholar
2023

Knowledge-in-Context: Towards Knowledgeable Semi-Parametric Language Models

ICLR 2023top-25%

Fully-parametric language models generally require a huge number of model parameters to store the necessary knowledge for solving multiple natural language tasks in zero/few-shot settings. In addition, it is hard to adapt to the evolving world knowledge without the costly model re-training. In this…

Cited by 20SourcePDFScholar
2023

PIVOINE: Instruction Tuning for Open-world Entity Profiling

EMNLP 2023long findings

This work considers the problem of Open-world Entity Profiling, a sub-domain of Open-world Information Extraction (Open-world IE). Unlike the conventional closed-world IE, Open-world IE is considered a more general situation where entities and relations could be beyond a predefined ontology. We seek…

Cited by 0SourceScholar
2023

Thrust: Adaptively Propels Large Language Models with External Knowledge

NeurIPS 2023poster

Although large-scale pre-trained language models (PTLMs) are shown to encode rich knowledge in their model parameters, the inherent knowledge in PTLMs can be opaque or static, making external knowledge necessary. However, the existing information retrieval techniques could be costly and may even int…

Cited by 11SourcePDFScholar
2023

Zemi: Learning Zero-Shot Semi-Parametric Language Models from Multiple Tasks

ACL 2023findings

Although large language models have exhibited impressive zero-shot ability, the huge model size generally incurs high cost. Recently, semi-parametric language models, which augment a smaller language model with retrieved related background knowledge, alleviate the need for storing everything into th…

2022

C-MORE: Pretraining to Answer Open-Domain Questions by Consulting Millions of References

ACL 2022short

We consider the problem of pretraining a two-stage open-domain question answering (QA) system (retriever + reader) with strong transfer capabilities. The key challenge is how to construct a large amount of high-quality question-answer-context triplets without task-specific annotations. Specifically,…

2022

Improving Machine Reading Comprehension with Contextualized Commonsense Knowledge

ACL 2022long

To perform well on a machine reading comprehension (MRC) task, machine readers usually require commonsense knowledge that is not explicitly mentioned in the given documents. This paper aims to extract a new kind of structured knowledge from scripts and use it to improve MRC. We focus on scripts as t…

2022

Learning-by-Narrating: Narrative Pre-Training for Zero-Shot Dialogue Comprehension

ACL 2022short

Comprehending a dialogue requires a model to capture diverse kinds of key information in the utterances, which are either scattered around or implicitly implied in different turns of conversations. Therefore, dialogue comprehension requires diverse capabilities such as paraphrasing, summarizing, and…

2022

Z-LaVI: Zero-Shot Language Solver Fueled by Visual Imagination

EMNLP 2022main

Large-scale pretrained language models have made significant advances in solving downstream language understanding tasks. However, they generally suffer from reporting bias, the phenomenon describing the lack of explicit commonsense knowledge in written text, e.g., ”an orange is orange”. To overcome…

2021

Connect-the-Dots: Bridging Semantics between Words and Definitions via Aligning Word Sense Inventories

EMNLP 2021main

Word Sense Disambiguation (WSD) aims to automatically identify the exact meaning of one word according to its context. Existing supervised models struggle to make correct predictions on rare word senses due to limited training data and can only select the best definition sentence from one predefined…

Cited by 10SourcePDFScholar
2020

Comprehensive Image Captioning via Scene Graph Decomposition

ECCV 2020poster

We address the challenging problem of image captioning by revisiting the representation of image scene graph. At the core of our method lies the decomposition of a scene graph into a set of sub-graphs, with each sub-graph capturing a semantic component of the input image. We design a deep model to s…

Cited by 160SourcePDFScholar
2020

TabFact: A Large-scale Dataset for Table-based Fact Verification

ICLR 2020poster

The problem of verifying whether a textual hypothesis holds based on the given evidence, also known as fact verification, plays an important role in the study of natural language understanding and semantic representation. However, existing studies are mainly restricted to dealing with unstructured e…

Cited by 517SourcecodeScholar
2020

Watch the Unobserved: A Simple Approach to Parallelizing Monte Carlo Tree Search

ICLR 2020talk

Monte Carlo Tree Search (MCTS) algorithms have achieved great success on many challenging benchmarks (e.g., Computer Go). However, they generally require a large number of rollouts, making their applications costly. Furthermore, it is also extremely challenging to parallelize MCTS due to its inheren…

Cited by 39SourcecodeScholar
2019

Stochastic Variance Reduced Primal Dual Algorithms for Empirical Composition Optimization

NeurIPS 2019poster

We consider a generic empirical composition optimization problem, where there are empirical averages present both outside and inside nonlinear loss functions. Such a problem is of interest in various machine learning applications, and cannot be directly solved by standard methods such as stochastic…

2019

Unsupervised Speech Recognition via Segmental Empirical Output Distribution Matching

ICLR 2019poster

We consider the problem of training speech recognition systems without using any labeled data, under the assumption that the learner can only access to the input utterances and a phoneme language model estimated from a non-overlapping corpus. We propose a fully unsupervised learning algorithm that a…

Cited by 48SourcePDFScholar
2018

Coupled Variational Bayes via Optimization Embedding

NeurIPS 2018poster

Variational inference plays a vital role in learning graphical models, especially on large-scale datasets. Much of its success depends on a proper choice of auxiliary distribution class for posterior approximation. However, how to pursue an auxiliary distribution class that achieves both good approx…

2018

M-Walk: Learning to Walk over Graphs using Monte Carlo Tree Search

NeurIPS 2018poster

Learning to walk over a graph towards a target node for a given query and a source node is an important problem in applications such as knowledge base completion (KBC). It can be formulated as a reinforcement learning (RL) problem with a known state transition model. To overcome the challenge of spa…

2018

SBEED: Convergent Reinforcement Learning with Nonlinear Function Approximation

ICML 2018oral

When function approximation is used, solving the Bellman optimality equation with stability guarantees has remained a major open problem in reinforcement learning for decades. The fundamental difficulty is that the Bellman operator may become an expansion in general, resulting in oscillating and eve…

Cited by 336SourcePDFScholar
2017

Character-level deep conflation for business data analytics

ICASSP 2017accepted

Connecting different text attributes associated with the same entity (conflation) is important in business data analytics since it could help merge two different tables in a database to provide a more comprehensive profile of an entity. However, the conflation task is challenging because two text st…

Cited by 0SourceScholar
2017

Q-LDA: Uncovering Latent Patterns in Text-based Sequential Decision Processes

NeurIPS 2017poster

In sequential decision making, it is often important and useful for end users to understand the underlying patterns or causes that lead to the corresponding decisions. However, typical deep reinforcement learning algorithms seldom provide such information due to their black-box nature. In this paper…

2017

Stochastic Variance Reduction Methods for Policy Evaluation

ICML 2017poster

Policy evaluation is concerned with estimating the value function that predicts long-term values of states under a given policy. It is a crucial step in many reinforcement-learning algorithms. In this paper, we focus on policy evaluation with linear function approximation over a fixed dataset. We fi…

Cited by 218SourcePDFScholar
2016

Interpreting the prediction process of a deep network constructed from supervised topic models

ICASSP 2016accepted

In this paper, we propose an approach to interpret the prediction process of the BP-sLDA model, which is a supervised Latent Dirichlet Allocation model trained by Back Propagation over a deep architecture. The model is shown to achieve state-of-the-art prediction performance on several large-scale t…

Cited by 0SourceScholar
2015

End-to-end Learning of LDA by Mirror-Descent Back Propagation over a Deep Architecture

NeurIPS 2015poster

We develop a fully discriminative learning approach for supervised Latent Dirichlet Allocation (LDA) model using Back Propagation (i.e., BP-sLDA), which maximizes the posterior probability of the prediction variable given the input document. Different from traditional variational learning or Gibbs s…