← Search

Hayeon Lee

10 accepted papers

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…

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…

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

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…

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

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

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

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