← Search

Jiaming Yang

5 accepted papers

2026

Rethinking KV Cache Eviction via a Unified Information-Theoretic Objective

ICML 2026poster

Key–value (KV) caching is essential for large language model inference, yet its memory overhead poses a critical bottleneck for long-context generation. Existing eviction policies predominantly rely on empirical heuristics, lacking a rigorous theoretical foundation. This work rethinks KV cache evict…

Cited by 0SourceScholar
2025

Visual Entity-Centric Prompting for Knowledge Retrieval in Knowledge-based VQA

ICASSP 2025accepted

External knowledge provides critical clues for knowledge-based visual question answering (KB-VQA), while the implicit knowledge in images is difficult to capture in order to construct effective queries for knowledge bases. To this end, we propose a visual entity-centric prompting for knowledge retri…

Cited by 0SourceScholar
2024

CharacterGLM: Customizing Social Characters with Large Language Models

EMNLP 2024industry

Character-based dialogue (CharacterDial) has become essential in the industry (e.g., Character.AI), enabling users to freely customize social characters for social interactions. However, the generalizability and adaptability across various conversational scenarios inherent in customizing social char…

Cited by 0SourcePDFScholar
2022

Pixelated Butterfly: Simple and Efficient Sparse training for Neural Network Models

ICLR 2022spotlight

Overparameterized neural networks generalize well but are expensive to train. Ideally one would like to reduce their computational cost while retaining their generalization benefits. Sparse model training is a simple and promising approach to achieve this, but there remain challenges as existing met…