← Search

Ke Ji

22 accepted papers

2026

DIVE: Scaling Diversity in Agentic Task Synthesis for Generalizable Tool Use

ICML 2026poster

Recent work increasingly synthesizes agentic tasks for post-training tool-using LLMs, yet robust generalization under shifts in tasks and toolsets remains an open challenge. We trace this brittleness to insufficient diversity in synthesized training tasks. Scaling diversity is difficult because trai…

Cited by 0SourceScholar
2025

CoD, Towards an Interpretable Medical Agent using Chain of Diagnosis

ACL 2025finding

The field of AI healthcare has undergone a significant transformation with the advent of large language models (LLMs), yet the challenges of interpretability within these models remain largely unaddressed. This study introduces **Chain-of-Diagnosis (CoD)** to enhance the interpretability of medical…

2025

Enhancing Persona Consistency for LLMs’ Role-Playing using Persona-Aware Contrastive Learning

ACL 2025finding

In recent years, large language models (LLMs) have achieved breakthrough progress in many dialogue generation tasks. However, their lack of emotion and fine-grained role awareness limits the model’s ability to provide personalized and diverse interactions further. Current methods face high costs in…

Cited by 0SourcePDFScholar
2025

QFFT, Question-Free Fine-Tuning for Adaptive Reasoning

NeurIPS 2025spotlight

Recent advancements in Long Chain-of-Thought (CoT) reasoning models have improved performance on complex tasks, but they suffer from overthinking, which generates redundant reasoning steps, especially for simple questions. This paper revisits the reasoning patterns of Long and Short CoT models, obse…

Cited by 0SourcecodeScholar
2025

RAG-Instruct: Boosting LLMs with Diverse Retrieval-Augmented Instructions

EMNLP 2025

Retrieval-Augmented Generation (RAG) has emerged as a key paradigm for enhancing large language models by incorporating external knowledge. However, current RAG methods exhibit limited capabilities in complex RAG scenarios and suffer from limited task diversity. To address these limitations, we prop

2025

SmartRAG: Jointly Learn RAG-Related Tasks From the Environment Feedback

ICLR 2025poster

RAG systems consist of multiple modules to work together. However, these modules are usually separately trained. We argue that a system like RAG that incorporates multiple modules should be jointly optimized to achieve optimal performance. To demonstrate this, we design a specific pipeline called Sm…

Cited by 4SourcePDFScholar
2025

The First Few Tokens Are All You Need: An Efficient and Effective Unsupervised Prefix Fine-Tuning Method for Reasoning Models

NeurIPS 2025poster

Improving the reasoning capabilities of large language models (LLMs) typically requires supervised fine-tuning with labeled data or computationally expensive sampling. We introduce Unsupervised Prefix Fine-Tuning (UPFT), which leverages the observation of Prefix Self-Consistency -- the shared initia…

Cited by 0SourceScholar
2025

Towards Medical Complex Reasoning with LLMs through Medical Verifiable Problems

ACL 2025finding

The breakthrough of OpenAI o1 highlights the potential of enhancing reasoning to improve LLM. Yet, most research in reasoning has focused on mathematical tasks, leaving domains like medicine underexplored. The medical domain, though distinct from mathematics, also demands robust reasoning to provide…

2025

Unlocking LLMs’ Self-Improvement Capacity with Autonomous Learning for Domain Adaptation

ACL 2025finding

Self-supervised pre-training and instruction fine-tuning demonstrate the potential of large language models (LLMs) for domain adaptation (DA). In pursuit of superhuman performance, LLMs have demonstrated significant potential in math and coding through self-improvement algorithms that rely on iterat…

2024

Domain-Hierarchy Adaptation via Chain of Iterative Reasoning for Few-shot Hierarchical Text Classification

IJCAI 2024poster

Recently, various pre-trained language models (PLMs) have been proposed to prove their impressive performances on a wide range of few-shot tasks. However, limited by the unstructured prior knowledge in PLMs, it is difficult to maintain consistent performance on complex hierarchically dependent tasks…

Cited by 1SourcePDFScholar
2024

Empirical Analysis of Dialogue Relation Extraction with Large Language Models

IJCAI 2024poster

Dialogue relation extraction (DRE) aims to extract relations between two arguments within a dialogue, which is more challenging than standard RE due to the higher person pronoun frequency and lower information density in dialogues. However, existing DRE methods still suffer from two serious issues:…

2024

Fast and Continual Knowledge Graph Embedding via Incremental LoRA

IJCAI 2024poster

Continual Knowledge Graph Embedding (CKGE) aims to efficiently learn new knowledge and simultaneously preserve old knowledge. Dominant approaches primarily focus on alleviating catastrophic forgetting of old knowledge but neglect efficient learning for the emergence of new knowledge. However, in rea…

2024

Incorporating Schema-Aware Description into Document-Level Event Extraction

IJCAI 2024poster

Document-level event extraction (DEE) aims to extract the structured event information from a given document, facing two critical challenges: (1) event arguments always scatter across sentences (arguments-scattering); (2) multiple events can co-occur in one document (multi-event). Most recent studie…

2024

LAMM: Label Alignment for Multi-Modal Prompt Learning

AAAI 2024technical

With the success of pre-trained visual-language (VL) models such as CLIP in visual representation tasks, transferring pre-trained models to downstream tasks has become a crucial paradigm. Recently, the prompt tuning paradigm, which draws inspiration from natural language processing (NLP), has made s…

2024

Meta In-Context Learning Makes Large Language Models Better Zero and Few-Shot Relation Extractors

IJCAI 2024poster

Relation extraction (RE) is an important task that aims to identify the relationships between entities in texts. While large language models (LLMs) have revealed remarkable in-context learning (ICL) capability for general zero and few-shot learning, recent studies indicate that current LLMs still st…

2024

OntoFact: Unveiling Fantastic Fact-Skeleton of LLMs via Ontology-Driven Reinforcement Learning

AAAI 2024technical

Large language models (LLMs) have demonstrated impressive proficiency in information retrieval, while they are prone to generating incorrect responses that conflict with reality, a phenomenon known as intrinsic hallucination. The critical challenge lies in the unclear and unreliable fact distributio…

2024

Recall, Retrieve and Reason: Towards Better In-Context Relation Extraction

IJCAI 2024poster

Relation extraction (RE) aims to identify relations between entities mentioned in texts. Although large language models (LLMs) have demonstrated impressive in-context learning (ICL) abilities in various tasks, they still suffer from poor performances compared to most supervised fine-tuned RE methods…

2024

Towards Continual Knowledge Graph Embedding via Incremental Distillation

AAAI 2024technical

Traditional knowledge graph embedding (KGE) methods typically require preserving the entire knowledge graph (KG) with significant training costs when new knowledge emerges. To address this issue, the continual knowledge graph embedding (CKGE) task has been proposed to train the KGE model by learning…

2024

Towards Injecting Medical Visual Knowledge into Multimodal LLMs at Scale

EMNLP 2024main

The rapid development of multimodal large language models (MLLMs), such as GPT-4V, has led to significant advancements. However, these models still face challenges in medical multimodal capabilities due to limitations in the quantity and quality of medical vision-text data, stemming from data privac…

2024

Unlocking Instructive In-Context Learning with Tabular Prompting for Relational Triple Extraction

COLING 2024main

The in-context learning (ICL) for relational triple extraction (RTE) has achieved promising performance, but still encounters two key challenges: (1) how to design effective prompts and (2) how to select proper demonstrations. Existing methods, however, fail to address these challenges appropriately…

Cited by 13SourcePDFScholar
2023

Hierarchical Verbalizer for Few-Shot Hierarchical Text Classification

ACL 2023long

Due to the complex label hierarchy and intensive labeling cost in practice, the hierarchical text classification (HTC) suffers a poor performance especially when low-resource or few-shot settings are considered. Recently, there is a growing trend of applying prompts on pre-trained language models (P…

2023

fmLRE: A Low-Resource Relation Extraction Model Based on Feature Mapping Similarity Calculation

AAAI 2023technical

Low-resource relation extraction (LRE) aims to extract relations from limited labeled corpora. Existing work takes advantages of self-training or distant supervision to expand the limited labeled data in the data-driven approaches, while the selection bias of pseudo labels may cause the error accum…