← Search

Bo Bai

5 accepted papers

2026

Scaling Knowledge Editing in LLMs to 100,000 Facts with Neural KV Database

ICLR 2026poster

Efficiently editing knowledge stored in Large Language Models (LLMs) enables model updates without large-scale training. One promising solution is Locate-and-Edit (L\&E), allowing simultaneous modifications of a massive number of factual knowledge. However, such editing may compromise the general ab…

Cited by 0SourcecodeScholar
2025

Efficient Prompt Compression with Evaluator Heads for Long-Context Transformer Inference

NeurIPS 2025spotlight

Although applications involving long-context inputs are crucial for the effective utilization of large language models (LLMs), they also result in increased computational costs and reduced performance. To address this challenge, we propose an efficient, training-free prompt compression method that r…

Cited by 0SourceScholar
2024

Extending Context Window of Large Language Models via Semantic Compression

ACL 2024findings

Transformer based Large Language Models (LLMs) often impose limitations on the length of the text input to ensure the generation of fluent and relevant responses due to the quadratic complexity. These constraints restrict their applicability in long text scenarios. In this paper, we propose a novel…

Cited by 24SourcePDFScholar
2021

Asymmetric Gained Deep Image Compression With Continuous Rate Adaptation

CVPR 2021poster

With the development of deep learning techniques, the combination of deep learning with image compression has drawn lots of attention. Recently, learned image compression methods had exceeded their classical counterparts in terms of rate-distortion performance. However, continuous rate adaptation re…

Cited by 162PDFScholar
2021

PartialFed: Cross-Domain Personalized Federated Learning via Partial Initialization

NeurIPS 2021poster

The burst of applications empowered by massive data have aroused unprecedented privacy concerns in AI society. Currently, data confidentiality protection has been one core issue during deep model training. Federated Learning (FL), which enables privacy-preserving training across multiple silos, gain…

Cited by 132SourcePDFScholar