← Search

Tei-Wei Kuo

8 accepted papers

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
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…

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…

2022

Speech Recovery For Real-World Self-Powered Intermittent Devices

ICASSP 2022accepted

The incompleteness of speech inputs severely degrades the performance of all the related speech signal processing applications. Although many researches have been proposed to address this issue, they controlled the data missing conditions by simulation with self-defined masking lengths or sizes. Bes…

Cited by 0SourceScholar
2021

PASSLEAF: A Pool-bAsed Semi-Supervised LEArning Framework for Uncertain Knowledge Graph Embedding

AAAI 2021technical

In this paper, we study the problem of embedding uncertain knowledge graphs, where each relation between entities is associated with a confidence score. Observing the existing embedding methods may discard the uncertainty information, only incorporate a specific type of score function, or cause many…

Cited by 29SourcePDFScholar
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
2020

Spatiotemporal Super-Resolution with Cross-Task Consistency and Its Semi-supervised Extension

IJCAI 2020poster

Spatiotemporal super-resolution (SR) aims to upscale both the spatial and temporal dimensions of input videos, and produces videos with higher frame resolutions and rates. It involves two essential sub-tasks: spatial SR and temporal SR. We design a two-stream network for spatiotemporal SR in this wo…