← Search

Yufei CUI

14 accepted papers

2026

Attention with Routed-Memory for Learnable Sparse Control

ICML 2026poster

Despite advances in long-context inference, large language models (LLMs) remain fundamentally limited by the key-value (KV) caching mechanisms that are necessary for stable computation. Management techniques, such as selective token eviction and pruning, have vastly mitigated the issues that have ar…

Cited by 0SourceScholar
2026

BAHOP: Similarity-based Basin Hopping for A fast hyper-parameter search in WSI classification

ICASSP 2026poster

Pre-processing whole slide images (WSIs) can impact classification performance. Our study shows that using fixed hyper-parameters for pre-processing out-of-domain WSIs can significantly degrade performance. Therefore, it is critical to search domain-specific hyper-parameters during inference. Howeve…

Cited by 0SourcePDFScholar
2026

Beyond Hard Writes and Rigid Preservation: Soft Recursive Least-Squares for Lifelong LLM Editing

IJCAI 2026

Model editing updates a pre-trained LLM with new facts or rules without retraining while preserving unrelated behavior. In real deployment, edits arrive as long streams, creating a plasticity–stability dilemma: repeated locate-then-edit “hard writes” can accumulate interference over time, while rigi

Cited by 0Scholar
2026

RAEE: A Robust Retrieval-Augmented Early Exit Framework for Efficient Inference

ICLR 2026poster

Deploying large language model inference remains challenging due to their high computational overhead. Early exit optimizes model inference by adaptively reducing the number of inference layers. Current methods typically train internal classifiers or use heuristic methods to determine the exit layer…

Cited by 0SourceScholar
2025

Advancing Multiple Instance Learning with Continual Learning for Whole Slide Imaging

CVPR 2025highlight

Advances in medical imaging and deep learning have propelled progress in whole slide image (WSI) analysis, with multiple instance learning (MIL) showing promise for efficient and accurate diagnostics. However, conventional MIL models often lack adaptability to evolving datasets, as they rely on stat…

Cited by 0SourcePDFScholar
2025

Mamba Modulation: On the Length Generalization of Mamba Models

NeurIPS 2025poster

The quadratic complexity of the attention mechanism in Transformer models has motivated the development of alternative architectures with sub-quadratic scaling, such as state-space models. Among these, Mamba has emerged as a leading architecture, achieving state-of-the-art results across a range of…

Cited by 0SourceScholar
2025

RALAD: Bridging the Real-to-Sim Domain Gap in Autonomous Driving with Retrieval-Augmented Learning

IROS 2025

As end-to-end autonomous driving advances toward real-world deployment, ensuring the safety of autonomous vehicles (AVs) has become a critical requirement for their commercial viability. While rule-based AVs have traditionally undergone rigorous testing in both real-world and simulated environments

Cited by 2SourcecodeScholar
2025

Transtreaming: Adaptive Delay-aware Transformer for Real-time Streaming Perception

AAAI 2025technical

Real-time object detection is critical for the decision-making process for many real-world applications, such as collision avoidance and path planning in autonomous driving. This work presents an innovative real-time streaming perception method, Transtreaming, which addresses the challenge of real-t…

2024

ReFusion: Improving Natural Language Understanding with Computation-Efficient Retrieval Representation Fusion

ICLR 2024poster

Retrieval-based augmentations (RA) incorporating knowledge from an external database into language models have greatly succeeded in various knowledge-intensive (KI) tasks. However, integrating retrievals in non-knowledge-intensive (NKI) tasks is still challenging. Existing works focus on concatenati…

2024

The Pitfalls and Promise of Conformal Inference Under Adversarial Attacks

ICML 2024poster

In safety-critical applications such as medical imaging and autonomous driving, where decisions have profound implications for patient health and road safety, it is imperative to maintain both high adversarial robustness to protect against potential adversarial attacks and reliable uncertainty quant…

2023

Bayes-MIL: A New Probabilistic Perspective on Attention-based Multiple Instance Learning for Whole Slide Images

ICLR 2023poster

Multiple instance learning (MIL) is a popular weakly-supervised learning model on the whole slide image (WSI) for AI-assisted pathology diagnosis. The recent advance in attention-based MIL allows the model to find its region-of-interest (ROI) for interpretation by learning the attention weights for…

Cited by 20SourcePDFScholar
2023

Retrieval-Augmented Multiple Instance Learning

NeurIPS 2023poster

Multiple Instance Learning (MIL) is a crucial weakly supervised learning method applied across various domains, e.g., medical diagnosis based on whole slide images (WSIs). Recent advancements in MIL algorithms have yielded exceptional performance when the training and test data originate from the sa…

2021

Bayesian Nested Neural Networks for Uncertainty Calibration and Adaptive Compression

CVPR 2021poster

Nested networks or slimmable networks are neural networks whose architectures can be adjusted instantly during testing time, e.g., based on computational constraints. Recent studies have focused on a "nested dropout" layer, which is able to order the nodes of a layer by importance during training, t…

Cited by 7PDFcodeScholar
2020

Fully Nested Neural Network for Adaptive Compression and Quantization

IJCAI 2020poster

Neural network compression and quantization are important tasks for fitting state-of-the-art models into the computational, memory and power constraints of mobile devices and embedded hardware. Recent approaches to model compression/quantization are based on reinforcement learning or search methods…

Cited by 0SourcePDFScholar