← Search

Hejie Cui

10 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
2026

T$^2$PO: Uncertainty-Guided Exploration Control for Stable Multi-Turn Agentic Reinforcement Learning

ICML 2026spotlight

Recent progress in multi-turn reinforcement learning (RL) has significantly improved reasoning LLMs' performances on complex interactive tasks. Despite advances in stabilization techniques such as fine-grained credit assignment and trajectory filtering, instability remains pervasive and often leads …

Cited by 0SourceScholar
2025

CLIMB: Data Foundations for Large Scale Multimodal Clinical Foundation Models

ICML 2025poster

Recent advances in clinical AI have enabled remarkable progress across many clinical domains. However, existing benchmarks and models are primarily limited to a small set of modalities and tasks, which hinders the development of large-scale multimodal methods that can make holistic assessments of pa…

2024

Biomedical Visual Instruction Tuning with Clinician Preference Alignment

NeurIPS 2024poster

Recent advancements in multimodal foundation models have showcased impressive capabilities in understanding and reasoning with visual and textual information. Adapting these foundation models trained for general usage to specialized domains like biomedicine requires large-scale domain-specific instr…

2024

Knowledge-Infused Prompting: Assessing and Advancing Clinical Text Data Generation with Large Language Models

ACL 2024findings

Clinical natural language processing faces challenges like complex medical terminology and clinical contexts. Recently, large language models (LLMs) have shown promise in this domain. Yet, their direct deployment can lead to privacy issues and are constrained by resources. To address this challenge,…

2024

Microstructures and Accuracy of Graph Recall by Large Language Models

NeurIPS 2024poster

Graphs data is crucial for many applications, and much of it exists in the relations described in textual format. As a result, being able to accurately recall and encode a graph described in earlier text is a basic yet pivotal ability that LLMs need to demonstrate if they are to perform reasoning ta…

2023

Neighborhood-Regularized Self-Training for Learning with Few Labels

AAAI 2023technical

Training deep neural networks (DNNs) with limited supervision has been a popular research topic as it can significantly alleviate the annotation burden. Self-training has been successfully applied in semi-supervised learning tasks, but one drawback of self-training is that it is vulnerable to the la…

2023

Open Visual Knowledge Extraction via Relation-Oriented Multimodality Model Prompting

NeurIPS 2023poster

Images contain rich relational knowledge that can help machines understand the world. Existing methods on visual knowledge extraction often rely on the pre-defined format (e.g., sub-verb-obj tuples) or vocabulary (e.g., relation types), restricting the expressiveness of the extracted knowledge. In t…

Cited by 6SourcePDFScholar
2023

PV2TEA: Patching Visual Modality to Textual-Established Information Extraction

ACL 2023findings

Information extraction, e.g., attribute value extraction, has been extensively studied and formulated based only on text. However, many attributes can benefit from image-based extraction, like color, shape, pattern, among others. The visual modality has long been underutilized, mainly due to multimo…