← Search

Sangmin Bae

15 accepted papers

2026

Composer: A Search Framework for Hybrid Neural Architecture Design

ICLR 2026poster

Hybrid model architectures that combine computational primitives (e.g., Attention, MLP) in different ratios have shown promising performance beyond Transformers. Some studies have shown that different interleavings of primitives can affect model quality as well. However, prior works explore the hybr…

Cited by 0SourceScholar
2025

Automated Filtering of Human Feedback Data for Aligning Text-to-Image Diffusion Models

ICLR 2025poster

Fine-tuning text-to-image diffusion models with human feedback is an effective method for aligning model behavior with human intentions. However, this alignment process often suffers from slow convergence due to the large size and noise present in human feedback datasets. In this work, we propose Fi…

2025

Mixture-of-Recursions: Learning Dynamic Recursive Depths for Adaptive Token-Level Computation

NeurIPS 2025poster

Scaling language models unlocks impressive capabilities, but the accompanying computational and memory demands make both training and deployment expensive. Existing efficiency efforts typically target either parameter sharing or adaptive computation, leaving open the question of how to attain both s…

Cited by 0SourceScholar
2025

MoHAVE: Mixture of Hierarchical Audio-Visual Experts for Robust Speech Recognition

ICML 2025poster

Audio-visual speech recognition (AVSR) has become critical for enhancing speech recognition in noisy environments by integrating both auditory and visual modalities. However, existing AVSR systems struggle to scale up without compromising computational efficiency. In this study, we introduce MoHAVE…

Cited by 2SourcePDFScholar
2025

Multi-Task Corrupted Prediction for Learning Robust Audio-Visual Speech Representation

ICLR 2025poster

Audio-visual speech recognition (AVSR) incorporates auditory and visual modalities to improve recognition accuracy, particularly in noisy environments where audio-only speech systems are insufficient. While previous research has largely addressed audio disruptions, few studies have dealt with visual…

2025

Relaxed Recursive Transformers: Effective Parameter Sharing with Layer-wise LoRA

ICLR 2025poster

Large language models (LLMs) are expensive to deploy. Parameter sharing offers a possible path towards reducing their size and cost, but its effectiveness in modern LLMs remains fairly limited. In this work, we revisit "layer tying" as form of parameter sharing in Transformers, and introduce novel m…

Cited by 5SourcePDFScholar
2024

Block Transformer: Global-to-Local Language Modeling for Fast Inference

NeurIPS 2024poster

We introduce the Block Transformer which adopts hierarchical global-to-local modeling to autoregressive transformers to mitigate the inference bottlenecks associated with self-attention. Self-attention requires the key-value (KV) cache of all previous sequences to be retrieved from memory at every d…

2024

Carpe diem: On the Evaluation of World Knowledge in Lifelong Language Models

NAACL 2024long

The dynamic nature of knowledge in an ever-changing world presents challenges for language models trained on static data; the model in the real world often requires not only acquiring new knowledge but also overwriting outdated information into updated ones. To study the ability of language models f…

2024

Hard Prompts Made Interpretable: Sparse Entropy Regularization for Prompt Tuning with RL

ACL 2024long

With the advent of foundation models, prompt tuning has positioned itself as an important technique for directing model behaviors and eliciting desired responses. Prompt tuning regards selecting appropriate keywords included into the input, thereby adapting to the downstream task without adjusting o…

2024

Stethoscope-Guided Supervised Contrastive Learning for Cross-Domain Adaptation on Respiratory Sound Classification

ICASSP 2024accepted

Despite the remarkable advances in deep learning technology, achieving satisfactory performance in lung sound classification remains a challenge due to the scarcity of available data. Moreover, the respiratory sound samples are collected from a variety of electronic stethoscopes, which could potenti…

Cited by 0SourceScholar
2023

Coreset Sampling From Open-Set for Fine-Grained Self-Supervised Learning

CVPR 2023poster

Deep learning in general domains has constantly been extended to domain-specific tasks requiring the recognition of fine-grained characteristics. However, real-world applications for fine-grained tasks suffer from two challenges: a high reliance on expert knowledge for annotation and necessity of a…

2023

Fast and Robust Early-Exiting Framework for Autoregressive Language Models with Synchronized Parallel Decoding

EMNLP 2023long main

To tackle the high inference latency exhibited by autoregressive language models, previous studies have proposed an early-exiting framework that allocates adaptive computation paths for each token based on the complexity of generating the subsequent token. However, we observed several shortcomings,…

Cited by 0SourcecodeScholar
2023

Re-Thinking Federated Active Learning Based on Inter-Class Diversity

CVPR 2023poster

Although federated learning has made awe-inspiring advances, most studies have assumed that the client's data are fully labeled. However, in a real-world scenario, every client may have a significant amount of unlabeled instances. Among the various approaches to utilizing unlabeled data, a federated…

2023

Self-Contrastive Learning: Single-Viewed Supervised Contrastive Framework Using Sub-network

AAAI 2023technical

Contrastive loss has significantly improved performance in supervised classification tasks by using a multi-viewed framework that leverages augmentation and label information. The augmentation enables contrast with another view of a single image but enlarges training time and memory usage. To exploi…

2022

Preservation of the Global Knowledge by Not-True Distillation in Federated Learning

NeurIPS 2022accept

In federated learning, a strong global model is collaboratively learned by aggregating clients' locally trained models. Although this precludes the need to access clients' data directly, the global model's convergence often suffers from data heterogeneity. This study starts from an analogy to contin…