← Search

Byeongho Heo

27 accepted papers

2026

Exploring Conditions for Diffusion Models in Robotic Control

CVPR 2026

While pre-trained visual representations have significantly advanced imitation learning, they are often task-agnostic as they remain frozen during policy learning. In this work, we explore leveraging pre-trained text-to-image diffusion models to obtain task-adaptive visual representations for roboti

Cited by 0SourceScholar
2026

MuCo: Multi-turn Contrastive Learning for Multimodal Embedding Model

CVPR 2026

Universal Multimodal embedding models built on Multimodal Large Language Models (MLLMs) have traditionally employed contrastive learning, which aligns representations of query-target pairs across different modalities. Yet, despite its empirical success, they are primarily built on a "single-turn" fo

Cited by 0SourcecodeScholar
2025

Masking meets Supervision: A Strong Learning Alliance

CVPR 2025poster

Pre-training with random masked inputs has emerged as a novel trend in self-supervised training. However, supervised learning still faces a challenge in adopting masking augmentations, primarily due to unstable training. In this paper, we propose a novel way to involve masking augmentations dubbed M…

2025

Token Bottleneck: One Token to Remember Dynamics

NeurIPS 2025poster

Deriving compact and temporally aware visual representations from dynamic scenes is essential for successful execution of sequential scene understanding tasks such as visual tracking and robotic manipulation. In this paper, we introduce Token Bottleneck (ToBo), a simple yet intuitive self-supervised…

Cited by 0SourcecodeScholar
2025

Token-Supervised Value Models for Enhancing Mathematical Problem-Solving Capabilities of Large Language Models

ICLR 2025poster

With the rapid advancement of test-time compute search strategies to improve the mathematical problem-solving capabilities of large language models (LLMs), the need for building robust verifiers has become increasingly important. However, all these inference strategies rely on existing verifiers ori…

Cited by 0SourcePDFScholar
2024

Learning with Unmasked Tokens Drives Stronger Vision Learners

ECCV 2024poster

"Masked image modeling (MIM) has become a leading self-supervised learning strategy. MIMs such as Masked Autoencoder (MAE) learn strong representations by randomly masking input tokens for the encoder to process, with the decoder reconstructing the masked tokens to the input. However, MIM pre-traine…

2024

Lipsum-FT: Robust Fine-Tuning of Zero-Shot Models Using Random Text Guidance

ICLR 2024poster

Large-scale contrastive vision-language pre-trained models provide the zero-shot model achieving competitive performance across a range of image classification tasks without requiring training on downstream data. Recent works have confirmed that while additional fine-tuning of the zero-shot model on…

2024

SeiT++: Masked Token Modeling Improves Storage-efficient Training

ECCV 2024poster

"Recent advancements in Deep Neural Network (DNN) models have significantly improved performance across computer vision tasks. However, achieving highly generalizable and high-performing vision models requires expansive datasets, resulting in significant storage requirements. This storage challenge…

2024

Similarity of Neural Architectures using Adversarial Attack Transferability

ECCV 2024poster

"In recent years, many deep neural architectures have been developed for image classification. Whether they are similar or dissimilar and what factors contribute to their (dis)similarities remains curious. To address this question, we aim to design a quantitative and scalable similarity measure betw…

Cited by 3SourcePDFScholar
2023

Scratching Visual Transformer's Back with Uniform Attention

ICCV 2023poster

The favorable performance of Vision Transformers (ViTs) is often attributed to the multi-head self-attention (MSA), which enables global interactions at each layer of a ViT model. Previous works acknowledge the property of long-range dependency for the effectiveness in MSA. In this work, we study t…

Cited by 39PDFScholar
2023

SeiT: Storage-Efficient Vision Training with Tokens Using 1% of Pixel Storage

ICCV 2023poster

We need billion-scale images to achieve more generalizable and ground-breaking vision models, as well as massive dataset storage to ship the images (e.g., the LAION-4B dataset needs 240TB storage space). However, it has become challenging to deal with unlimited dataset storage with limited storage i…

Cited by 10PDFcodeScholar
2023

What Do Self-Supervised Vision Transformers Learn?

ICLR 2023poster

We present a comparative study on how and why contrastive learning (CL) and masked image modeling (MIM) differ in their representations and in their performance of downstream tasks. In particular, we demonstrate that self-supervised Vision Transformers (ViTs) have the following properties: (1) CL tr…

2022

Improving Ensemble Distillation With Weight Averaging and Diversifying Perturbation

ICML 2022spotlight

Ensembles of deep neural networks have demonstrated superior performance, but their heavy computational cost hinders applying them for resource-limited environments. It motivates distilling knowledge from the ensemble teacher into a smaller student network, and there are two important design choices…

2022

Joint Global and Local Hierarchical Priors for Learned Image Compression

CVPR 2022poster

Recently, learned image compression methods have outperformed traditional hand-crafted ones including BPG. One of the keys to this success is learned entropy models that estimate the probability distribution of the quantized latent representation. Like other vision tasks, most recent learned entropy…

Cited by 95PDFcodeScholar
2022

K-Centered Patch Sampling for Efficient Video Recognition

ECCV 2022poster

"For decades, it has been a common practice to choose a subset of video frames for reducing the computational burden of a video understanding model. In this paper, we argue that this popular heuristic might be sub-optimal under recent transformer-based models. Specifically, inspired by that transfor…

2022

The Majority Can Help the Minority: Context-Rich Minority Oversampling for Long-Tailed Classification

CVPR 2022poster

The problem of class imbalanced data is that the generalization performance of the classifier deteriorates due to the lack of data from minority classes. In this paper, we propose a novel minority over-sampling method to augment diversified minority samples by leveraging the rich context of the majo…

Cited by 200PDFcodeScholar
2022

ViDT: An Efficient and Effective Fully Transformer-based Object Detector

ICLR 2022poster

Transformers are transforming the landscape of computer vision, especially for recognition tasks. Detection transformers are the first fully end-to-end learning systems for object detection, while vision transformers are the first fully transformer-based architecture for image classification. In thi…

2021

AdamP: Slowing Down the Slowdown for Momentum Optimizers on Scale-invariant Weights

ICLR 2021poster

Normalization techniques, such as batch normalization (BN), are a boon for modern deep learning. They let weights converge more quickly with often better generalization performances. It has been argued that the normalization-induced scale invariance among the weights provides an advantageous ground…

2021

Re-Labeling ImageNet: From Single to Multi-Labels, From Global to Localized Labels

CVPR 2021poster

ImageNet has been the most popular image classification benchmark, but it is also the one with a significant level of label noise. Recent studies have shown that many samples contain multiple classes, despite being assumed to be a single-label benchmark. They have thus proposed to turn ImageNet eval…

Cited by 183PDFcodeScholar
2021

Rethinking Spatial Dimensions of Vision Transformers

ICCV 2021poster

Vision Transformer (ViT) extends the application range of transformers from language processing to computer vision tasks as being an alternative architecture against the existing convolutional neural networks (CNN). Since the transformer-based architecture has been innovative for computer vision mod…

Cited by 761PDFcodeScholar
2021

Show, Attend and Distill: Knowledge Distillation via Attention-based Feature Matching

AAAI 2021technical

Knowledge distillation extracts general knowledge from a pretrained teacher network and provides guidance to a target student network. Most studies manually tie intermediate features of the teacher and student, and transfer knowledge through predefined links. However, manual selection often construc…

2019

A Comprehensive Overhaul of Feature Distillation

ICCV 2019poster

We investigate the design aspects of feature distillation methods achieving network compression and propose a novel feature distillation method in which the distillation loss is designed to make a synergy among various aspects: teacher transform, student transform, distillation feature position and…

Cited by 791PDFcodeScholar