← Search

Jae W. Lee

11 accepted papers

2026

Libra: Effective yet Efficient Load Balancing for Large-scale MoE Inference

ICLR 2026poster

Distributed inference of large-scale Mixture-of-Experts (MoE) models faces a critical challenge: expert load imbalance. Numerous system-level approaches have been proposed for load balancing, but they either fail to achieve a satisfactory level of balance or introduce new bottlenecks due to the over…

Cited by 0SourceScholar
2026

SpareTrain: Fault-Tolerant LLM Training via Low-Cost Dual Modular Redundancy

ICLR 2026poster

Dual Modular Redundancy (DMR) is a highly effective mechanism for detecting silent data corruption (SDC)—a critical reliability concern in large language model (LLM) training—by executing each operation twice. However, its high computation overhead has prevented practical deployment at scale. In thi…

Cited by 0SourceScholar
2025

DP-LLM: Runtime Model Adaptation with Dynamic Layer-wise Precision Assignment

NeurIPS 2025poster

How can we effectively handle queries for on-device large language models (LLMs) with varying runtime constraints, such as latency and accuracy? Multi-scale quantization addresses this challenge by enabling memory-efficient runtime model adaptation of LLMs through the overlaying of multiple model va…

Cited by 0SourceScholar
2025

FastPoint: Accelerating 3D Point Cloud Model Inference via Sample Point Distance Prediction

ICCV 2025poster

Deep neural networks have revolutionized 3D point cloud processing, yet efficiently handling large and irregular point clouds remains challenging. To tackle this problem, we introduce FastPoint, a novel software-based acceleration technique that leverages the predictable distance trend between sampl…

2025

FlashTP: Fused, Sparsity-Aware Tensor Product for Machine Learning Interatomic Potentials

ICML 2025spotlight

Machine Learning Interatomic Potentials (MLIPs) enable efficient molecular dynamics (MD) simulations with high accuracy. While equivariant MLIPs achieve state-of-the-art accuracy, they face significant computational bottlenecks centered around their Tensor-Product layer, which account for up to 75\%…

2025

GuidedQuant: Large Language Model Quantization via Exploiting End Loss Guidance

ICML 2025poster

Post-training quantization is a key technique for reducing the memory and inference latency of large language models by quantizing weights and activations without requiring retraining. However, existing methods either (1) fail to account for the varying importance of hidden features to the end loss…

2025

KVzip: Query-Agnostic KV Cache Compression with Context Reconstruction

NeurIPS 2025oral

Transformer-based large language models (LLMs) cache context as key-value (KV) pairs during inference. As context length grows, KV cache sizes expand, leading to substantial memory overhead and increased attention latency. This paper introduces \textit{KVzip}, a query-agnostic KV cache eviction meth…

Cited by 0SourcecodeScholar
2025

NestedFP: High-Performance, Memory-Efficient Dual-Precision Floating Point Support for LLMs

NeurIPS 2025poster

Meeting service-level objectives (SLOs) in Large Language Models (LLMs) serving is critical, but managing the high variability in load presents a significant challenge. Recent advancements in FP8 inference, backed by native hardware support, offer a potential solution: executing FP16 models by defau…

Cited by 0SourcecodeScholar
2024

Any-Precision LLM: Low-Cost Deployment of Multiple, Different-Sized LLMs

ICML 2024oral

Recently, considerable efforts have been directed towards compressing Large Language Models (LLMs), which showcase groundbreaking capabilities across diverse applications but entail significant deployment costs due to their large sizes. Meanwhile, much less attention has been given to mitigating the…

2023

Not All Neighbors Matter: Point Distribution-Aware Pruning for 3D Point Cloud

AAAI 2023technical

Applying deep neural networks to 3D point cloud processing has demonstrated a rapid pace of advancement in those domains where 3D geometry information can greatly boost task performance, such as AR/VR, robotics, and autonomous driving. However, as the size of both the neural network model and 3D poi…

2022

"L3: Accelerator-Friendly Lossless Image Format for High-Resolution, High-Throughput DNN Training"

ECCV 2022poster

"The training process of deep neural networks (DNNs) is usually pipelined with stages for data preparation on CPUs followed by gradient computation on accelerators like GPUs. In an ideal pipeline, the end-to-end training throughput is eventually limited by the throughput of the accelerator, not by t…