← Search

Sung Ju Hwang

148 accepted papers

2026

Avatar Forcing: Real-Time Interactive Head Avatar Generation for Natural Conversation

CVPR 2026

Talking head generation creates lifelike avatars from static portraits for virtual communication and content creation. However, current models do not yet convey the feeling of truly interactive communication, often generating one-way responses that lack emotional engagement. We identify two key chal

Cited by 0SourcecodeScholar
2026

Frame Guidance: Training-Free Guidance for Frame-Level Control in Video Diffusion Model

ICLR 2026poster

Advancements in diffusion models have significantly improved video quality, directing attention to fine-grained controllability. However, many existing methods depend on fine-tuning large-scale video models for specific tasks, which becomes increasingly impractical as model sizes continue to grow. I…

Cited by 0SourcecodeScholar
2026

LS-Merge: Merging Language Models in Latent Space

ICLR 2026poster

Model merging in weight space is an efficient way to reuse pretrained models, but existing methods typically assume matching architectures or sizes, making heterogeneous merges brittle or infeasible. We address this limitation by encoding model weights into a smooth latent space, enabling cross-arch…

Cited by 0SourcecodeScholar
2026

Multi-View Encoders for Performance Prediction in LLM-Based Agentic Workflows

ICLR 2026poster

Large language models (LLMs) have demonstrated remarkable capabilities across diverse tasks, but optimizing LLM-based agentic systems remains challenging due to the vast search space of agent configurations, prompting strategies, and communication patterns. Existing approaches often rely on heuristi…

Cited by 0SourceScholar
2026

Multimodal Prompt Optimization: Why Not Leverage Multiple Modalities for MLLMs

ICLR 2026poster

Large Language Models (LLMs) have shown remarkable success, and their multimodal expansions (MLLMs) further unlock capabilities spanning images, videos, and other modalities beyond text. However, despite this shift, prompt optimization approaches, designed to reduce the burden of manual prompt craft…

Cited by 0SourcecodeScholar
2026

Paper2Code: Automating Code Generation from Scientific Papers in Machine Learning

ICLR 2026poster

Despite the rapid growth of machine learning research, corresponding code implementations are often unavailable, making it slow and labor-intensive for researchers to reproduce results and build upon prior work. In the meantime, recent Large Language Models (LLMs) excel at understanding scientific d…

Cited by 0SourcecodeScholar
2025

A Training-Free Sub-quadratic Cost Transformer Model Serving Framework with Hierarchically Pruned Attention

ICLR 2025poster

In modern large language models (LLMs), increasing the context length is crucial for improving comprehension and coherence in long-context, multi-modal, and retrieval-augmented language generation. While many recent transformer models attempt to extend their context length over a million tokens, th…

Cited by 0SourcePDFScholar
2025

AutoML-Agent: A Multi-Agent LLM Framework for Full-Pipeline AutoML

ICML 2025poster

Automated machine learning (AutoML) accelerates AI development by automating tasks in the development pipeline, such as optimal model search and hyperparameter tuning. Existing AutoML systems often require technical expertise to set up complex tools, which is in general time-consuming and requires a…

2025

Bayesian Neural Scaling Law Extrapolation with Prior-Data Fitted Networks

ICML 2025poster

Scaling has been a major driver of recent advancements in deep learning. Numerous empirical studies have found that scaling laws often follow the power-law and proposed several variants of power-law functions to predict the scaling behavior at larger scales. However, existing methods mostly rely on…

Cited by 0SourcePDFScholar
2025

Cost-Sensitive Freeze-thaw Bayesian Optimization for Efficient Hyperparameter Tuning

NeurIPS 2025poster

In this paper, we address the problem of cost-sensitive hyperparameter optimization (HPO) built upon freeze-thaw Bayesian optimization (BO). Specifically, we assume a scenario where users want to early-stop the HPO process when the expected performance improvement is not satisfactory with respect to…

Cited by 0SourceScholar
2025

Database-Augmented Query Representation for Information Retrieval

EMNLP 2025

Information retrieval models that aim to search for documents relevant to a query have shown multiple successes, which have been applied to diverse tasks. Yet, the query from the user is oftentimes short, which challenges the retrievers to correctly fetch relevant documents. To tackle this, previous

2025

Delta Attention: Fast and Accurate Sparse Attention Inference by Delta Correction

NeurIPS 2025poster

The attention mechanism of a transformer has a quadratic complexity, leading to high inference costs and latency for long sequences. However, attention matrices are mostly sparse, which implies that many entries may be omitted from computation for efficient inference. Sparse attention inference meth…

Cited by 0SourceScholar
2025

Diffusion-based Neural Network Weights Generation

ICLR 2025poster

Transfer learning is a cornerstone of modern deep learning, yet it remains constrained by challenges in model selection and the overhead of extensive model storage. In this work, we present Diffusion-based Neural Network Weights Generation, D2NWG, a novel framework that leverages diffusion processes…

2025

Distilling LLM Agent into Small Models with Retrieval and Code Tools

NeurIPS 2025spotlight

Large language models (LLMs) excel at complex reasoning tasks but remain computationally expensive, limiting their practical deployment. To address this, recent works have focused on distilling reasoning capabilities into smaller language models (sLMs) using chain-of-thought (CoT) traces from teache…

Cited by 0SourcecodeScholar
2025

Efficient Long Context Language Model Retrieval with Compression

ACL 2025long

Long Context Language Models (LCLMs) have emerged as a new paradigm to perform Information Retrieval (IR), which enables the direct ingestion and retrieval of information by processing an entire corpus in their single context, showcasing the potential to surpass traditional sparse and dense retrieva…

2025

FedSVD: Adaptive Orthogonalization for Private Federated Learning with LoRA

NeurIPS 2025poster

Low-Rank Adaptation (LoRA), which introduces a product of two trainable low-rank matrices into frozen pre-trained weights, is widely used for efficient fine-tuning of language models in federated learning (FL). However, when combined with differentially private stochastic gradient descent (DP-SGD),…

Cited by 0SourceScholar
2025

HarmAug: Effective Data Augmentation for Knowledge Distillation of Safety Guard Models

ICLR 2025poster

Safety guard models that detect malicious queries aimed at large language models (LLMs) are essential for ensuring the secure and responsible deployment of LLMs in real-world applications. However, deploying existing safety guard models with billions of parameters alongside LLMs on mobile devices is…

2025

Learning Diverse Attacks on Large Language Models for Robust Red-Teaming and Safety Tuning

ICLR 2025poster

Red-teaming, or identifying prompts that elicit harmful responses, is a critical step in ensuring the safe and responsible deployment of large language models (LLMs). Developing effective protection against many modes of attack prompts requires discovering diverse attacks. Automated red-teaming typi…

2025

Mol-LLaMA: Towards General Understanding of Molecules in Large Molecular Language Model

NeurIPS 2025poster

Understanding molecules is key to understanding organisms and driving advances in drug discovery, requiring interdisciplinary knowledge across chemistry and biology. Although large molecular language models have achieved notable success in task transfer, they often struggle to accurately analyze mol…

Cited by 0SourcecodeScholar
2025

ResearchAgent: Iterative Research Idea Generation over Scientific Literature with Large Language Models

NAACL 2025long

The pace of scientific research, vital for improving human life, is complex, slow, and needs specialized expertise. Meanwhile, novel, impactful research often stems from both a deep understanding of prior work, and a cross-pollination of ideas across domains and fields. To enhance the productivity o…

2025

SafeRoute: Adaptive Model Selection for Efficient and Accurate Safety Guardrails in Large Language Models

ACL 2025finding

Deploying large language models (LLMs) in real-world applications requires robust safety guard models to detect and block harmful user prompts. While large safety guard models achieve strong performance, their computational cost is substantial. To mitigate this, smaller distilled models are used, bu…

2025

Silent Branding Attack: Trigger-free Data Poisoning Attack on Text-to-Image Diffusion Models

CVPR 2025poster

Text-to-image diffusion models have achieved remarkable success in generating high-quality contents from text prompts. However, their reliance on publicly available data and the growing trend of data sharing for fine-tuning make these models particularly vulnerable to data poisoning attacks. In this…

Cited by 1SourcePDFScholar
2025

Sketch-of-Thought: Efficient LLM Reasoning with Adaptive Cognitive-Inspired Sketching

EMNLP 2025

Recent advances in large language models (LLMs) have enabled strong reasoning capabilities through Chain-of-Thought (CoT) prompting, which elicits step-by-step problem solving, but often at the cost of excessive verbosity in intermediate outputs, leading to increased computational overhead. We propo

2025

Training Free Exponential Context Extension via Cascading KV Cache

ICLR 2025poster

The transformer's context window is vital for tasks such as few-shot learning and conditional generation as it preserves previous tokens for active memory. However, as the context lengths increase, the computational costs grow quadratically, hindering the deployment of large language models (LLMs) i…

2025

VideoICL: Confidence-based Iterative In-context Learning for Out-of-Distribution Video Understanding

CVPR 2025poster

Recent advancements in video large multimodal models (LMMs) have significantly improved their video understanding and reasoning capabilities. However, their performance drops on out-of-distribution (OOD) tasks that are underrepresented in training data. Traditional methods like fine-tuning on OOD da…

2025

VideoRAG: Retrieval-Augmented Generation over Video Corpus

ACL 2025finding

Retrieval-Augmented Generation (RAG) is a powerful strategy for improving the factual accuracy of models by retrieving external knowledge relevant to queries and incorporating it into the generation process. However, existing approaches primarily focus on text, with some recent advancements consider…

2024

Adaptive-RAG: Learning to Adapt Retrieval-Augmented Large Language Models through Question Complexity

NAACL 2024long

Retrieval-Augmented Large Language Models (LLMs), which incorporate the non-parametric knowledge from external knowledge bases into LLMs, have emerged as a promising approach to enhancing response accuracy in several tasks, such as Question-Answering (QA). However, even though there are various appr…

2024

BECoTTA: Input-dependent Online Blending of Experts for Continual Test-time Adaptation

ICML 2024poster

Continual Test-Time Adaptation (CTTA) is designed to optimize the model during deployment under changing conditions. CTTA is an important problem as it enables models to remain effective and reliable in dynamic and evolving environments. However, tackling the CTTA problem is nontrivial. The model ne…

Cited by 11SourcePDFScholar
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

Concept-skill Transferability-based Data Selection for Large Vision-Language Models

EMNLP 2024main

Instruction tuning, or supervised finetuning on extensive task-specific data, is necessary for Large Vision-Language Models (LVLMs) to generalize well across a broad range of vision-language (VL) tasks. However, training on large VL datasets can become prohibitively expensive. In this work, we intro…

2024

DiffusionNAG: Predictor-guided Neural Architecture Generation with Diffusion Models

ICLR 2024poster

Existing NAS methods suffer from either an excessive amount of time for repetitive sampling and training of many task-irrelevant architectures. To tackle such limitations of existing NAS methods, we propose a paradigm shift from NAS to a novel conditional Neural Architecture Generation (NAG) framewo…

2024

ECLIPSE: Efficient Continual Learning in Panoptic Segmentation with Visual Prompt Tuning

CVPR 2024poster

Panoptic segmentation combining semantic and instance segmentation stands as a cutting-edge computer vision task. Despite recent progress with deep learning models the dynamic nature of real-world applications necessitates continual learning where models adapt to new classes (plasticity) over time w…

2024

EVEREST: Efficient Masked Video Autoencoder by Removing Redundant Spatiotemporal Tokens

ICML 2024poster

Masked Video Autoencoder (MVA) approaches have demonstrated their potential by significantly outperforming previous video representation learning methods. However, they waste an excessive amount of computations and memory in predicting uninformative tokens/frames due to random masking strategies. (e…

Cited by 7SourcePDFScholar
2024

Generative Modeling on Manifolds Through Mixture of Riemannian Diffusion Processes

ICML 2024poster

Learning the distribution of data on Riemannian manifolds is crucial for modeling data from non-Euclidean space, which is required by many applications in diverse scientific fields. Yet, existing generative models on manifolds suffer from expensive divergence computation or rely on approximations of…

2024

Identity Decoupling for Multi-Subject Personalization of Text-to-Image Models

NeurIPS 2024poster

Text-to-image diffusion models have shown remarkable success in generating personalized subjects based on a few reference images. However, current methods often fail when generating multiple subjects simultaneously, resulting in mixed identities with combined attributes from different subjects. In t…

Cited by 12SourcePDFScholar
2024

KOALA: Empirical Lessons Toward Memory-Efficient and Fast Diffusion Models for Text-to-Image Synthesis

NeurIPS 2024poster

As text-to-image (T2I) synthesis models increase in size, they demand higher inference costs due to the need for more expensive GPUs with larger memory, which makes it challenging to reproduce these models in addition to the restricted access to training datasets. Our study aims to reduce these infe…

Cited by 2SourcePDFScholar
2024

Latent Paraphrasing: Perturbation on Layers Improves Knowledge Injection in Language Models

NeurIPS 2024poster

As Large Language Models (LLMs) are increasingly deployed in specialized domains with continuously evolving knowledge, the need for timely and precise knowledge injection has become essential. Fine-tuning with paraphrased data is a common approach to enhance knowledge injection, yet it faces two sig…

2024

One Prompt is not Enough: Automated Construction of a Mixture-of-Expert Prompts

ICML 2024poster

Large Language Models (LLMs) exhibit strong generalization capabilities to novel tasks when prompted with language instructions and in-context demos. Since this ability sensitively depends on the quality of prompts, various methods have been explored to automate the instruction design. While these m…

2024

Progressive Fourier Neural Representation for Sequential Video Compilation

ICLR 2024poster

Neural Implicit Representation (NIR) has recently gained significant attention due to its remarkable ability to encode complex and high-dimensional data into representation space and easily reconstruct it through a trainable mapping function. However, NIR methods assume a one-to-one mapping between…

Cited by 2SourcePDFScholar
2024

SEA: Sparse Linear Attention with Estimated Attention Mask

ICLR 2024poster

The transformer architecture has driven breakthroughs in recent years on tasks which require modeling pairwise relationships between sequential elements, as is the case in natural language understanding. However, long seqeuences pose a problem due to the quadratic complexity of the attention operati…

2024

STELLA: Continual Audio-Video Pre-training with SpatioTemporal Localized Alignment

ICML 2024poster

Continuously learning a variety of audio-video semantics over time is crucial for audio-related reasoning tasks in our ever-evolving world. However, this is a nontrivial problem and poses two critical challenges: sparse spatio-temporal correlation between audio-video pairs and multimodal correlation…

Cited by 4SourcePDFScholar
2024

Self-Supervised Dataset Distillation for Transfer Learning

ICLR 2024poster

Dataset distillation aims to optimize a small set so that a model trained on the set achieves performance similar to that of a model trained on the full dataset. While many supervised methods have achieved remarkable success in distilling a large dataset into a small set of representative samples, h…

2024

Set-based Neural Network Encoding Without Weight Tying

NeurIPS 2024poster

We propose a neural network weight encoding method for network property prediction that utilizes set-to-set and set-to-vector functions to efficiently encode neural network parameters. Our approach is capable of encoding neural networks in a model zoo of mixed architecture and different parameter si…

Cited by 0SourcePDFScholar
2023

A Study on Knowledge Distillation from Weak Teacher for Scaling Up Pre-trained Language Models

ACL 2023findings

Distillation from Weak Teacher (DWT) is a method of transferring knowledge from a smaller, weaker teacher model to a larger student model to improve its performance. Previous studies have shown that DWT can be effective in the vision domain and natural language processing (NLP) pre-training stage. S…

2023

Co-training and Co-distillation for Quality Improvement and Compression of Language Models

EMNLP 2023long findings

Knowledge Distillation (KD) compresses computationally expensive pre-trained language models (PLMs) by transferring their knowledge to smaller models, allowing their use in resource-constrained or real-time settings. However, most smaller models fail to surpass the performance of the original larger…

Cited by 0SourceScholar
2023

Direct Fact Retrieval from Knowledge Graphs without Entity Linking

ACL 2023long

There has been a surge of interest in utilizing Knowledge Graphs (KGs) for various natural language processing/understanding tasks. The conventional mechanism to retrieve facts in KGs usually involves three steps: entity span detection, entity disambiguation, and relation classification. However, th…

2023

Effective Targeted Attacks for Adversarial Self-Supervised Learning

NeurIPS 2023poster

Recently, unsupervised adversarial training (AT) has been highlighted as a means of achieving robustness in models without any label information. Previous studies in unsupervised AT have mostly focused on implementing self-supervised learning (SSL) frameworks, which maximize the instance-wise classi…

Cited by 4SourcePDFScholar
2023

Exploring Chemical Space with Score-based Out-of-distribution Generation

ICML 2023poster

A well-known limitation of existing molecular generative models is that the generated molecules highly resemble those in the training set. To generate truly novel molecules that may have even better properties for de novo drug discovery, more powerful exploration in the chemical space is necessary.…

2023

Exploring The Role of Mean Teachers in Self-supervised Masked Auto-Encoders

ICLR 2023poster

Masked image modeling (MIM) has become a popular strategy for self-supervised learning (SSL) of visual representations with Vision Transformers. A representative MIM model, the masked auto-encoder (MAE), randomly masks a subset of image patches and reconstructs the masked patches given the unmasked…

2023

Generalizable Lightweight Proxy for Robust NAS against Diverse Perturbations

NeurIPS 2023poster

Recent neural architecture search (NAS) frameworks have been successful in finding optimal architectures for given conditions (e.g., performance or latency). However, they search for optimal architectures in terms of their performance on clean images only, while robustness against various types of p…

2023

Grad-StyleSpeech: Any-Speaker Adaptive Text-to-Speech Synthesis with Diffusion Models

ICASSP 2023accepted

There has been a significant progress in Text-To-Speech (TTS) synthesis technology in recent years, thanks to the advancement in neural generative modeling. However, existing methods on any-speaker adaptive TTS have achieved unsatisfactory performance, due to their suboptimal accuracy in mimicking t…

Cited by 0SourceScholar
2023

Knowledge-Augmented Language Model Verification

EMNLP 2023long main

Recent Language Models (LMs) have shown impressive capabilities in generating texts with the knowledge internalized in parameters. Yet, LMs often generate the factually incorrect responses to the given queries, since their knowledge may be inaccurate, incomplete, and outdated. To address this proble…

Cited by 0SourcecodeScholar
2023

Knowledge-Augmented Reasoning Distillation for Small Language Models in Knowledge-Intensive Tasks

NeurIPS 2023poster

Large Language Models (LLMs) have shown promising performance in knowledge-intensive reasoning tasks that require a compound understanding of knowledge. However, deployment of the LLMs in real-world applications can be challenging due to their high computational requirements and concerns on data pr…

2023

Language Detoxification with Attribute-Discriminative Latent Space

ACL 2023long

Transformer-based Language Models (LMs) have achieved impressive results on natural language understanding tasks, but they can also generate toxic text such as insults, threats, and profanity, limiting their real-world applications. To overcome this issue, a few text generation approaches aim to det…

2023

Meta-prediction Model for Distillation-Aware NAS on Unseen Datasets

ICLR 2023top-25%

Distillation-aware Neural Architecture Search (DaNAS) aims to search for an optimal student architecture that obtains the best performance and/or efficiency when distilling the knowledge from a given teacher model. Previous DaNAS methods have mostly tackled the search for the neural architecture for…

2023

On the Soft-Subnetwork for Few-Shot Class Incremental Learning

ICLR 2023poster

Inspired by Regularized Lottery Ticket Hypothesis, which states that competitive smooth (non-binary) subnetworks exist within a dense network, we propose a few-shot class-incremental learning method referred to as Soft-SubNetworks (SoftNet). Our objective is to learn a sequence of sessions increment…

2023

Personalized Subgraph Federated Learning

ICML 2023poster

Subgraphs of a larger global graph may be distributed across multiple devices, and only locally accessible due to privacy restrictions, although there may be links between subgraphs. Recently proposed subgraph Federated Learning (FL) methods deal with those missing links across local subgraphs while…

2023

Phrase Retrieval for Open Domain Conversational Question Answering with Conversational Dependency Modeling via Contrastive Learning

ACL 2023findings

Open-Domain Conversational Question Answering (ODConvQA) aims at answering questions through a multi-turn conversation based on a retriever-reader pipeline, which retrieves passages and then predicts answers with them. However, such a pipeline approach not only makes the reader vulnerable to the err…

2023

STXD: Structural and Temporal Cross-Modal Distillation for Multi-View 3D Object Detection

NeurIPS 2023poster

3D object detection (3DOD) from multi-view images is an economically appealing alternative to expensive LiDAR-based detectors, but also an extremely challenging task due to the absence of precise spatial cues. Recent studies have leveraged the teacher-student paradigm for cross-modal distillation, w…

Cited by 8SourcePDFScholar
2023

Scalable Set Encoding with Universal Mini-Batch Consistency and Unbiased Full Set Gradient Approximation

ICML 2023poster

Recent work on mini-batch consistency (MBC) for set functions has brought attention to the need for sequentially processing and aggregating chunks of a partitioned set while guaranteeing the same output for all partitions. However, existing constraints on MBC architectures lead to models with limite…

2023

Self-Distillation for Further Pre-training of Transformers

ICLR 2023poster

Pre-training a large transformer model on a massive amount of unlabeled data and fine-tuning it on labeled datasets for diverse downstream tasks has proven to be a successful strategy, for a variety of vision and natural language processing tasks. However, direct fine-tuning of the pre-trained model…

Cited by 10SourcePDFScholar
2023

Self-Supervised Set Representation Learning for Unsupervised Meta-Learning

ICLR 2023poster

Unsupervised meta-learning (UML) essentially shares the spirit of self-supervised learning (SSL) in that their goal aims at learning models without any human supervision so that the models can be adapted to downstream tasks. Further, the learning objective of self-supervised learning, which pulls po…

Cited by 11SourcePDFScholar
2023

Test-Time Self-Adaptive Small Language Models for Question Answering

EMNLP 2023short findings

Recent instruction-finetuned large language models (LMs) have achieved notable performances in various tasks, such as question-answering (QA). However, despite their ability to memorize a vast amount of general knowledge across diverse tasks, they might be suboptimal on specific tasks due to their l…

Cited by 0SourcecodeScholar
2023

Text-Conditioned Sampling Framework for Text-to-Image Generation with Masked Generative Models

ICCV 2023poster

Token-based masked generative models are gaining popularity for their fast inference time with parallel decoding. While recent token-based approaches achieve competitive performance to diffusion-based models, their generation performance is still suboptimal as they sample multiple tokens simultaneou…

Cited by 5PDFScholar
2023

The Devil Is in the Points: Weakly Semi-Supervised Instance Segmentation via Point-Guided Mask Representation

CVPR 2023poster

In this paper, we introduce a novel learning scheme named weakly semi-supervised instance segmentation (WSSIS) with point labels for budget-efficient and high-performance instance segmentation. Namely, we consider a dataset setting consisting of a few fully-labeled images and a lot of point-labeled…

2022

Augmenting Document Representations for Dense Retrieval with Interpolation and Perturbation

ACL 2022short

Dense retrieval models, which aim at retrieving the most relevant document for an input query on a dense representation space, have gained considerable attention for their remarkable success. Yet, dense models require a vast amount of labeled training data for notable performance, whereas it is ofte…

2022

Bitwidth Heterogeneous Federated Learning with Progressive Weight Dequantization

ICML 2022spotlight

In practical federated learning scenarios, the participating devices may have different bitwidths for computation and memory storage by design. However, despite the progress made in device-heterogeneous federated learning scenarios, the heterogeneity in the bitwidth specifications in the hardware ha…

Cited by 29SourcePDFScholar
2022

Consistency Regularization for Adversarial Robustness

AAAI 2022technical

Adversarial training (AT) is currently one of the most successful methods to obtain the adversarial robustness of deep neural networks. However, the phenomenon of robust overfitting, i.e., the robustness starts to decrease significantly during AT, has been problematic, not only making practitioners…

2022

Factorized-FL: Personalized Federated Learning with Parameter Factorization & Similarity Matching

NeurIPS 2022accept

In real-world federated learning scenarios, participants could have their own personalized labels incompatible with those from other clients, due to using different label permutations or tackling completely different tasks or domains. However, most existing FL approaches cannot effectively tackle su…

2022

Forget-free Continual Learning with Winning Subnetworks

ICML 2022spotlight

Inspired by Lottery Ticket Hypothesis that competitive subnetworks exist within a dense network, we propose a continual learning method referred to as Winning SubNetworks (WSN), which sequentially learns and selects an optimal subnetwork for each task. Specifically, WSN jointly learns the model weig…

2022

Graph Self-supervised Learning with Accurate Discrepancy Learning

NeurIPS 2022accept

Self-supervised learning of graph neural networks (GNNs) aims to learn an accurate representation of the graphs in an unsupervised manner, to obtain transferable representations of them for diverse downstream tasks. Predictive learning and contrastive learning are the two most prevalent approaches f…

2022

Learning to Generate Inversion-Resistant Model Explanations

NeurIPS 2022accept

The wide adoption of deep neural networks (DNNs) in mission-critical applications has spurred the need for interpretable models that provide explanations of the model's decisions. Unfortunately, previous studies have demonstrated that model explanations facilitate information leakage, rendering DNN…

Cited by 3SourcePDFScholar
2022

MPViT: Multi-Path Vision Transformer for Dense Prediction

CVPR 2022poster

Dense computer vision tasks such as object detection and segmentation require effective multi-scale feature representation for detecting or classifying objects or regions with varying sizes. While Convolutional Neural Networks (CNNs) have been the dominant architectures for such tasks, recently intr…

Cited by 360PDFcodeScholar
2022

Meta Learning Low Rank Covariance Factors for Energy Based Deterministic Uncertainty

ICLR 2022poster

Numerous recent works utilize bi-Lipschitz regularization of neural network layers to preserve relative distances between data instances in the feature spaces of each layer. This distance sensitivity with respect to the data aids in tasks such as uncertainty calibration and out-of-distribution (OOD)…

Cited by 2SourcePDFScholar
2022

Online Coreset Selection for Rehearsal-based Continual Learning

ICLR 2022poster

A dataset is a shred of crucial evidence to describe a task. However, each data point in the dataset does not have the same potential, as some of the data points can be more representative or informative than others. This unequal importance among the data points may have a large impact in rehearsal-…

Cited by 180SourcePDFScholar
2022

Online Hyperparameter Meta-Learning with Hypergradient Distillation

ICLR 2022spotlight

Many gradient-based meta-learning methods assume a set of parameters that do not participate in inner-optimization, which can be considered as hyperparameters. Although such hyperparameters can be optimized using the existing gradient-based hyperparameter optimization (HO) methods, they suffer from…

Cited by 9SourcePDFScholar
2022

Representational Continuity for Unsupervised Continual Learning

ICLR 2022oral

Continual learning (CL) aims to learn a sequence of tasks without forgetting the previously acquired knowledge. However, recent CL advances are restricted to supervised continual learning (SCL) scenarios. Consequently, they are not scalable to real-world applications where the data distribution is o…

2022

Saliency Grafting: Innocuous Attribution-Guided Mixup with Calibrated Label Mixing

AAAI 2022technical

The Mixup scheme suggests mixing a pair of samples to create an augmented training sample and has gained considerable attention recently for improving the generalizability of neural networks. A straightforward and widely used extension of Mixup is to combine with regional dropout-like methods: remov…

Cited by 25SourcePDFScholar
2022

Score-based Generative Modeling of Graphs via the System of Stochastic Differential Equations

ICML 2022spotlight

Generating graph-structured data requires learning the underlying distribution of graphs. Yet, this is a challenging problem, and the previous graph generative methods either fail to capture the permutation-invariance property of graphs or cannot sufficiently model the complex dependency between nod…

2022

Sequential Reptile: Inter-Task Gradient Alignment for Multilingual Learning

ICLR 2022poster

Multilingual models jointly pretrained on multiple languages have achieved remarkable performance on various multilingual downstream tasks. Moreover, models finetuned on a single monolingual downstream task have shown to generalize to unseen languages. In this paper, we first show that it is crucial…

Cited by 16SourcePDFScholar
2022

Set-based Meta-Interpolation for Few-Task Meta-Learning

NeurIPS 2022accept

Meta-learning approaches enable machine learning systems to adapt to new tasks given few examples by leveraging knowledge from related tasks. However, a large number of meta-training tasks are still required for generalization to unseen tasks during meta-testing, which introduces a critical bottlen…

Cited by 9SourcePDFScholar
2021

Accurate Learning of Graph Representations with Graph Multiset Pooling

ICLR 2021poster

Graph neural networks have been widely used on modeling graph data, achieving impressive results on node classification and link prediction tasks. Yet, obtaining an accurate representation for a graph further requires a pooling function that maps a set of node representations into a compact form. A…

2021

Clinical Risk Prediction with Temporal Probabilistic Asymmetric Multi-Task Learning

AAAI 2021technical

Although recent multi-task learning methods have shown to be effective in improving the generalization of deep neural networks, they should be used with caution for safety-critical applications, such as clinical risk prediction. This is because even if they achieve improved task-average performance,…

2021

Cluster-Promoting Quantization With Bit-Drop for Minimizing Network Quantization Loss

ICCV 2021poster

Network quantization, which aims to reduce the bit-lengths of the network weights and activations, has emerged for their deployments to resource-limited devices. Although recent studies have successfully discretized a full-precision network, they still incur large quantization errors after training,…

Cited by 15PDFScholar
2021

Contrastive Learning with Adversarial Perturbations for Conditional Text Generation

ICLR 2021poster

Recently, sequence-to-sequence (seq2seq) models with the Transformer architecture have achieved remarkable performance on various conditional text generation tasks, such as machine translation. However, most of them are trained with teacher forcing with the ground truth label given at each time step…

2021

Edge Representation Learning with Hypergraphs

NeurIPS 2021poster

Graph neural networks have recently achieved remarkable success in representing graph-structured data, with rapid progress in both the node embedding and graph pooling methods. Yet, they mostly focus on capturing information from the nodes considering their connectivity, and not much work has been d…

2021

FedMix: Approximation of Mixup under Mean Augmented Federated Learning

ICLR 2021poster

Federated learning (FL) allows edge devices to collectively learn a model without directly sharing data within each device, thus preserving privacy and eliminating the need to store data globally. While there are promising results under the assumption of independent and identically distributed (iid)…

Cited by 229SourcePDFScholar
2021

Federated Continual Learning with Weighted Inter-client Transfer

ICML 2021spotlight

There has been a surge of interest in continual learning and federated learning, both of which are important in deep neural networks in real-world scenarios. Yet little research has been done regarding the scenario where each client learns on a sequence of tasks from a private local data stream. Thi…

2021

Federated Semi-Supervised Learning with Inter-Client Consistency & Disjoint Learning

ICLR 2021poster

While existing federated learning approaches mostly require that clients have fully-labeled data to train on, in realistic settings, data obtained at the client-side often comes without any accompanying labels. Such deficiency of labels may result from either high labeling cost, or difficulty of ann…

2021

GTA: Graph Truncated Attention for Retrosynthesis

AAAI 2021technical

Retrosynthesis is the task of predicting reactant molecules from a given product molecule and is, important in organic chemistry because the identification of a synthetic path is as demanding as the discovery of new chemical compounds. Recently, the retrosynthesis task has been solved automatically…

Cited by 71SourcePDFScholar
2021

Hardware-adaptive Efficient Latency Prediction for NAS via Meta-Learning

NeurIPS 2021spotlight

For deployment, neural architecture search should be hardware-aware, in order to satisfy the device-specific constraints (e.g., memory usage, latency and energy consumption) and enhance the model efficiency. Existing methods on hardware-aware NAS collect a large number of samples (e.g., accuracy and…

2021

Hit and Lead Discovery with Explorative RL and Fragment-based Molecule Generation

NeurIPS 2021poster

Recently, utilizing reinforcement learning (RL) to generate molecules with desired properties has been highlighted as a promising strategy for drug design. Molecular docking program -- a physical simulation that estimates protein-small molecule binding affinity -- can be an ideal reward scoring func…

Cited by 78SourcePDFScholar
2021

Large-Scale Meta-Learning with Continual Trajectory Shifting

ICML 2021spotlight

Meta-learning of shared initialization parameters has shown to be highly effective in solving few-shot learning tasks. However, extending the framework to many-shot scenarios, which may further enhance its practicality, has been relatively overlooked due to the technical difficulties of meta-learnin…

Cited by 19SourcePDFScholar
2021

Learning to Perturb Word Embeddings for Out-of-distribution QA

ACL 2021long

QA models based on pretrained language models have achieved remarkable performance on various benchmark datasets. However, QA models do not generalize well to unseen data that falls outside the training distribution, due to distributional shifts. Data augmentation (DA) techniques which drop/replace…

2021

Learning to Sample with Local and Global Contexts in Experience Replay Buffer

ICLR 2021poster

Experience replay, which enables the agents to remember and reuse experience from the past, has played a significant role in the success of off-policy reinforcement learning (RL). To utilize the experience replay efficiently, the existing sampling methods allow selecting out more meaningful experien…

Cited by 26SourcePDFScholar
2021

Meta-GMVAE: Mixture of Gaussian VAE for Unsupervised Meta-Learning

ICLR 2021spotlight

Unsupervised learning aims to learn meaningful representations from unlabeled data which can captures its intrinsic structure, that can be transferred to downstream tasks. Meta-learning, whose objective is to learn to generalize across tasks such that the learned model can rapidly adapt to a novel t…

2021

Meta-StyleSpeech : Multi-Speaker Adaptive Text-to-Speech Generation

ICML 2021spotlight

With rapid progress in neural text-to-speech (TTS) models, personalized speech generation is now in high demand for many applications. For practical applicability, a TTS model should generate high-quality speech with only a few audio samples from the given speaker, that are also short in length. How…

2021

Mini-Batch Consistent Slot Set Encoder for Scalable Set Encoding

NeurIPS 2021poster

Most existing set encoding algorithms operate under the implicit assumption that all the set elements are accessible, and that there are ample computational and memory resources to load the set into memory during training and inference. However, both assumptions fail when the set is excessively lar…

Cited by 8SourcePDFScholar
2021

Rapid Neural Architecture Search by Learning to Generate Graphs from Datasets

ICLR 2021poster

Despite the success of recent Neural Architecture Search (NAS) methods on various tasks which have shown to output networks that largely outperform human-designed networks, conventional NAS methods have mostly tackled the optimization of searching for the network architecture for a single task (data…

2021

RetCL: A Selection-based Approach for Retrosynthesis via Contrastive Learning

IJCAI 2021poster

Retrosynthesis, of which the goal is to find a set of reactants for synthesizing a target product, is an emerging research area of deep learning. While the existing approaches have shown promising results, they currently lack the ability to consider availability (e.g., stability or purchasability) o…

Cited by 23SourcePDFScholar
2021

Task-Adaptive Neural Network Search with Meta-Contrastive Learning

NeurIPS 2021spotlight

Most conventional Neural Architecture Search (NAS) approaches are limited in that they only generate architectures without searching for the optimal parameters. While some NAS methods handle this issue by utilizing a supernet trained on a large-scale dataset such as ImageNet, they may be suboptimal…

2020

Adversarial Neural Pruning with Latent Vulnerability Suppression

ICML 2020poster

Despite the remarkable performance of deep neural networks on various computer vision tasks, they are known to be susceptible to adversarial perturbations, which makes it challenging to deploy them in real-world safety-critical applications. In this paper, we conjecture that the leading cause of adv…

2020

Attribution Preservation in Network Compression for Reliable Network Interpretation

NeurIPS 2020poster

Neural networks embedded in safety-sensitive applications such as self-driving cars and wearable health monitors rely on two important techniques: input attribution for hindsight analysis and network compression to reduce its size for edge-computing. In this paper, we show that these seemingly unrel…

Cited by 11SourcePDFScholar
2020

Bootstrapping neural processes

NeurIPS 2020poster

Unlike in the traditional statistical modeling for which a user typically hand-specify a prior, Neural Processes (NPs) implicitly define a broad class of stochastic processes with neural networks. Given a data stream, NP learns a stochastic process that best describes the data. While this ``data-dri…

2020

Cost-Effective Interactive Attention Learning with Neural Attention Processes

ICML 2020poster

We propose a novel interactive learning framework which we refer to as Interactive Attention Learning (IAL), in which the human supervisors interactively manipulate the allocated attentions, to correct the model’s behaviour by updating the attention-generating network. However, such a model is prone…

2020

Distribution Aligning Refinery of Pseudo-label for Imbalanced Semi-supervised Learning

NeurIPS 2020poster

While semi-supervised learning (SSL) has proven to be a promising way for leveraging unlabeled data when labeled data is scarce, the existing SSL algorithms typically assume that training class distributions are balanced. However, these SSL algorithms trained under imbalanced class distributions can…

2020

Few-shot Visual Reasoning with Meta-Analogical Contrastive Learning

NeurIPS 2020poster

While humans can solve a visual puzzle that requires logical reasoning by observing only few samples, it would require training over a large number of samples for state-of-the-art deep reasoning models to obtain similar performance on the same task. In this work, we propose to solve such a few-shot…

Cited by 29SourcePDFScholar
2020

Learning to Balance: Bayesian Meta-Learning for Imbalanced and Out-of-distribution Tasks

ICLR 2020talk

While tasks could come with varying the number of instances and classes in realistic settings, the existing meta-learning approaches for few-shot classification assume that number of instances per task and class is fixed. Due to such restriction, they learn to equally utilize the meta-knowledge acro…

Cited by 144SourcecodeScholar
2020

Learning to Extrapolate Knowledge: Transductive Few-shot Out-of-Graph Link Prediction

NeurIPS 2020poster

Many practical graph problems, such as knowledge graph construction and drug-drug interaction prediction, require to handle multi-relational graphs. However, handling real-world multi-relational graphs with Graph Neural Networks (GNNs) is often challenging due to their evolving nature, as new entiti…

2020

Meta Dropout: Learning to Perturb Latent Features for Generalization

ICLR 2020poster

A machine learning model that generalizes well should obtain low errors on unseen test examples. Thus, if we know how to optimally perturb training examples to account for test examples, we may achieve better generalization performance. However, obtaining such perturbation is not possible in standar…

Cited by 60SourcecodeScholar
2020

Meta Variance Transfer: Learning to Augment from the Others

ICML 2020poster

Humans have the ability to robustly recognize objects with various factors of variations such as nonrigid transformations, background noises, and changes in lighting conditions. However, training deep learning models generally require huge amount of data instances under diverse variations, to ensure…

Cited by 60SourcePDFScholar
2020

MetaPerturb: Transferable Regularizer for Heterogeneous Tasks and Architectures

NeurIPS 2020spotlight

Regularization and transfer learning are two popular techniques to enhance model generalization on unseen data, which is a fundamental problem of machine learning. Regularization techniques are versatile, as they are task- and architecture-agnostic, but they do not exploit a large amount of data ava…

2020

Scalable and Order-robust Continual Learning with Additive Parameter Decomposition

ICLR 2020poster

While recent continual learning methods largely alleviate the catastrophic problem on toy-sized datasets, there are issues that remain to be tackled in order to apply them to real-world problem domains. First, a continual learning model should effectively handle catastrophic forgetting and be effici…

Cited by 200SourcecodeScholar
2020

Why Not to Use Zero Imputation? Correcting Sparsity Bias in Training Neural Networks

ICLR 2020poster

Handling missing data is one of the most fundamental problems in machine learning. Among many approaches, the simplest and most intuitive way is zero imputation, which treats the value of a missing entry simply as zero. However, many studies have experimentally confirmed that zero imputation results…

Cited by 47SourcecodeScholar
2019

LEARNING TO PROPAGATE LABELS: TRANSDUCTIVE PROPAGATION NETWORK FOR FEW-SHOT LEARNING

ICLR 2019poster

The goal of few-shot learning is to learn a classifier that generalizes well even when trained with a limited number of training instances per class. The recently introduced meta-learning approaches tackle this problem by learning a generic classifier across a large number of multiclass classificati…

2019

Learning to Quantize Deep Networks by Optimizing Quantization Intervals With Task Loss

CVPR 2019oral

Reducing bit-widths of activations and weights of deep networks makes it efficient to compute and store them in memory, which is crucial in their deployments to resource-limited devices, such as mobile phones. However, decreasing bit-widths with quantization generally yields drastically degraded acc…

Cited by 476PDFScholar
2018

Joint Active Feature Acquisition and Classification with Variable-Size Set Encoding

NeurIPS 2018poster

We consider the problem of active feature acquisition where the goal is to sequentially select the subset of features in order to achieve the maximum prediction performance in the most cost-effective way at test time. In this work, we formulate this active feature acquisition as a jointly learning p…

2018

Uncertainty-Aware Attention for Reliable Interpretation and Prediction

NeurIPS 2018poster

Attention mechanism is effective in both focusing the deep learning models on relevant features and interpreting them. However, attentions may be unreliable since the networks that generate them are often trained in a weakly-supervised manner. To overcome this limitation, we introduce the notion of…

2017

SplitNet: Learning to Semantically Split Deep Networks for Parameter Reduction and Model Parallelization

ICML 2017poster

We propose a novel deep neural network that is both lightweight and effectively structured for model parallelization. Our network, which we name as SplitNet, automatically learns to split the network weights into either a set or a hierarchy of multiple groups that use disjoint sets of features, by l…

Cited by 101SourcePDFScholar
2015

Expanding Object Detector's Horizon: Incremental Learning Framework for Object Detection in Videos

CVPR 2015poster

Over the last several years it has been shown that image-based object detectors are sensitive to the training data and often fail to generalize to examples that fall outside the original training sample domain (e.g., videos). A number of domain adaptation (DA) techniques have been proposed to add…

Cited by 58SourcePDFScholar