← Search

Chun Jason Xue

14 accepted papers

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

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

A2ATS: Retrieval-Based KV Cache Reduction via Windowed Rotary Position Embedding and Query-Aware Vector Quantization

ACL 2025finding

Long context large language models (LLMs) pose significant challenges for efficient serving due to the large memory footprint and high access overhead of KV cache.Retrieval-based KV cache reduction methods can mitigate these challenges, typically by offloading the complete KV cache to CPU and retrie…

2025

MLWQ: Efficient Small Language Model Deployment via Multi-Level Weight Quantization

EMNLP 2025

Small language models (SLMs) are gaining attention for their lower computational and memory needs while maintaining strong performance. However, efficiently deploying SLMs on resource-constrained devices remains a significant challenge. Post-training quantization(PTQ) is a widely used compression te

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
2024

BehaviorGPT: Smart Agent Simulation for Autonomous Driving with Next-Patch Prediction

NeurIPS 2024poster

Simulating realistic behaviors of traffic agents is pivotal for efficiently validating the safety of autonomous driving systems. Existing data-driven simulators primarily use an encoder-decoder architecture to encode the historical trajectories before decoding the future. However, the heterogeneity…

Cited by 18SourcePDFScholar
2024

CHESS: Optimizing LLM Inference via Channel-Wise Thresholding and Selective Sparsification

EMNLP 2024main

Deploying large language models (LLMs) on edge devices presents significant challenges due to the substantial computational overhead and memory requirements. Activation sparsification can mitigate these resource challenges by reducing the number of activated neurons during inference. Existing method…

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

When Compression Meets Model Compression: Memory-Efficient Double Compression for Large Language Models

EMNLP 2024finding

Large language models (LLMs) exhibit excellent performance in various tasks. However, the memory requirements of LLMs present a great challenge when deploying on memory-limited devices, even for quantized LLMs. This paper introduces a framework to compress LLM after quantization further, achieving a…

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