← Search

Jushi Kai

6 accepted papers

2026

FreqKV: Key-Value Compression in Frequency Domain for Context Window Extension

ICLR 2026poster

Existing key-value (KV) cache compression methods for large language models (LLMs) often rely on token eviction, which risks losing critical local information in both long prefilling and decoding scenarios. When extrapolating beyond the pretrained context length, their performance degrades sharply o…

Cited by 0SourcecodeScholar
2026

MLP Memory: A Retriever-Pretrained Memory for Large Language Models

ICLR 2026poster

Modern approaches to enhancing Large Language Models' factual accuracy and knowledge utilization face a fundamental trade-off: non-parametric retrieval-augmented generation (RAG) provides flexible access to external knowledge but suffers from high inference latency and shallow integration, while par…

Cited by 0SourcecodeScholar
2024

ItiNera: Integrating Spatial Optimization with Large Language Models for Open-domain Urban Itinerary Planning

EMNLP 2024industry

Citywalk, a recently popular form of urban travel, requires genuine personalization and understanding of fine-grained requests compared to traditional itinerary planning. In this paper, we introduce the novel task of Open-domain Urban Itinerary Planning (OUIP), which generates personalized urban iti…

2024

Leveraging Grammar Induction for Language Understanding and Generation

EMNLP 2024finding

Grammar induction has made significant progress in recent years. However, it is not clear how the application of induced grammar could enhance practical performance in downstream tasks. In this work, we introduce an unsupervised grammar induction method for language understanding and generation. We…

2024

SH2: Self-Highlighted Hesitation Helps You Decode More Truthfully

EMNLP 2024finding

Large language models (LLMs) demonstrate great performance in text generation. However, LLMs are still suffering from hallucinations. In this work, we propose an inference-time method, Self-Highlighted Hesitation (SH2), to help LLMs decode more truthfully. SH2 is based on a simple fact rooted in inf…

2022

Syntax-guided Localized Self-attention by Constituency Syntactic Distance

EMNLP 2022finding

Recent works have revealed that Transformers are implicitly learning the syntactic information in its lower layers from data, albeit is highly dependent on the quality and scale of the training data. However, learning syntactic information from data is not necessary if we can leverage an external sy…