← Search

Yue Niu

4 accepted papers

2026

HitKV: Activation Frequency Knows Which Tokens Are Important

AAAI 2026technical

The demand for long-context processing in large language models (LLMs) continues to escalate alongside rapid advancements in their capabilities. However, the intermediate attention keys and values (KV cache) employed to avoid re-computations, also grow linearly with sequence length, far exceeding th

Cited by 0SourcePDFScholar
2025

MobiZO: Enabling Efficient LLM Fine-Tuning at the Edge via Inference Engines

EMNLP 2025

Large Language Models (LLMs) are currently pre-trained and fine-tuned on large cloud servers. The next frontier is LLM personalization, where a foundation model can be fine-tuned with user/task-specific data. Given the sensitive nature of such private data, it is desirable to fine-tune these models

2024

All Rivers Run to the Sea: Private Learning with Asymmetric Flows

CVPR 2024poster

Data privacy is of great concern in cloud machine-learning service platforms when sensitive data are exposed to service providers. While private computing environments (e.g. secure enclaves) and cryptographic approaches (e.g. homomorphic encryption) provide strong privacy protection their computing…

Cited by 1SourcePDFScholar
2024

Ethos: Rectifying Language Models in Orthogonal Parameter Space

NAACL 2024findings

Language models (LMs) have greatly propelled the research on natural language processing. However, LMs also raise concerns regarding the generation of biased or toxic content and the potential disclosure of private information from the training dataset. In this work, we present a new efficient appro…