← Search

Pengcheng Jiang

14 accepted papers

2026

Efficient, Property-Aligned Fan-Out Retrieval via RL-Amortized Diffusion

ICML 2026poster

Many modern retrieval problems are \emph{set-valued}: given a broad intent, the system must return a \emph{collection} of results that optimizes higher-order properties (e.g., diversity, coverage, complementarity, coherence) while staying grounded to a fixed database. Set-valued objectives are inher…

Cited by 0SourceScholar
2026

GRACE: Generative Representation Learning via Contrastive Policy Optimization

ICLR 2026poster

Prevailing methods for training Large Language Models (LLMs) as text encoders rely on contrastive losses that treat the model as a black-box function, discarding its generative and reasoning capabilities in favor of static embeddings. We introduce \GRACE{} (Generative Representation Learning via Con…

Cited by 0SourcecodeScholar
2026

RAS: Retrieval-And-Structuring for Knowledge-Intensive LLM Generation

ICLR 2026poster

Large language models (LLMs) have achieved impressive performance on knowledge-intensive tasks, yet they often struggle with multi-step reasoning due to the unstructured nature of retrieved context. While retrieval-augmented generation (RAG) methods provide external information, the lack of explicit…

Cited by 0SourcecodeScholar
2026

Rethinking the Reranker: Boundary-Aware Evidence Selection for Robust Retrieval-Augmented Generation

ICML 2026poster

Retrieval-Augmented Generation (RAG) systems remain brittle under realistic retrieval noise, even when the required evidence appears in the top-$K$ results. A key reason is that retrievers and rerankers optimize solely for relevance, often selecting either trivial, answer-revealing passages or evide…

Cited by 0SourceScholar
2025

Bi-level Contrastive Learning for Knowledge-Enhanced Molecule Representations

AAAI 2025technical

Molecular representation learning is vital for various downstream applications, including the analysis and prediction of molecular properties and side effects. While Graph Neural Networks (GNNs) have been a popular framework for modeling molecular data, they often struggle to capture the full comple…

Cited by 2SourcePDFScholar
2025

Reasoning-Enhanced Healthcare Predictions with Knowledge Graph Community Retrieval

ICLR 2025poster

Large language models (LLMs) have demonstrated significant potential in clinical decision support. Yet LLMs still suffer from hallucinations and lack fine-grained contextual medical knowledge, limiting their high-stake healthcare applications such as clinical diagnosis. Traditional retrieval-augment…

2025

Topic Coverage-based Demonstration Retrieval for In-Context Learning

EMNLP 2025

The effectiveness of in-context learning relies heavily on selecting demonstrations that provide all the necessary information for a given test input.To achieve this, it is crucial to identify and cover fine-grained knowledge requirements. However, prior methods often retrieve demonstrations based s

2025

s3: You Don’t Need That Much Data to Train a Search Agent via RL

EMNLP 2025

Retrieval-augmented generation (RAG) systems empower large language models (LLMs) to access external knowledge during inference. Recent advances have enabled LLMs to act as search agents via reinforcement learning (RL), improving information acquisition through multi-turn interactions with retrieval

2024

GenRES: Rethinking Evaluation for Generative Relation Extraction in the Era of Large Language Models

NAACL 2024long

The field of relation extraction (RE) is experiencing a notable shift towards generative relation extraction (GRE), leveraging the capabilities of large language models (LLMs). However, we discovered that traditional relation extraction (RE) metrics like precision and recall fall short in evaluating…

2024

GraphCare: Enhancing Healthcare Predictions with Personalized Knowledge Graphs

ICLR 2024poster

Clinical predictive models often rely on patients’ electronic health records (EHR), but integrating medical knowledge to enhance predictions and decision-making is challenging. This is because personalized predictions require personalized knowledge graphs (KGs), which are difficult to generate from…

Cited by 39SourcePDFScholar
2024

KG-FIT: Knowledge Graph Fine-Tuning Upon Open-World Knowledge

NeurIPS 2024poster

Knowledge Graph Embedding (KGE) techniques are crucial in learning compact representations of entities and relations within a knowledge graph, facilitating efficient reasoning and knowledge discovery. While existing methods typically focus either on training KGE models solely based on graph structur…

2024

Taxonomy-guided Semantic Indexing for Academic Paper Search

EMNLP 2024main

Academic paper search is an essential task for efficient literature discovery and scientific advancement. While dense retrieval has advanced various ad-hoc searches, it often struggles to match the underlying academic concepts between queries and documents, which is critical for paper search. To ena…

2024

TriSum: Learning Summarization Ability from Large Language Models with Structured Rationale

NAACL 2024long

The advent of large language models (LLMs) has significantly advanced natural language processing tasks like text summarization. However, their large size and computational demands, coupled with privacy concerns in data transmission, limit their use in resource-constrained and privacy-centric settin…

2023

Text Augmented Open Knowledge Graph Completion via Pre-Trained Language Models

ACL 2023findings

The mission of open knowledge graph (KG) completion is to draw new findings from known facts. Existing works that augment KG completion require either (1) factual triples to enlarge the graph reasoning space or (2) manually designed prompts to extract knowledge from a pre-trained language model (PLM…