Fine-Tuning Masked Diffusion for Provable Self-Correction

Jaeyeon Kim*, Seunggeun Kim*, Taekyun Lee*, David Z. Pan, Hyeji Kim, Sham Kakade, Sitan Chen (* equal contribution)

ICML 2026
ICLR Workshop MM Intelligence 2026

[Paper] [Code]

Efficient Weighted Sampling via Score-based Generative Models

Heasung Kim*, Taekyun Lee*, Hyeji Kim, Gustavo de Veciana (* equal contribution)

CVPR 2026

[Paper] [Video]

Learning Ray-Tracing for Radio Propagation via Cross-Attention-Based Diffusion Models

Juseong Park, Taekyun Lee, Yunchou Xing, Jie Chen, Amitava Ghosh, Jeffrey G. Andrews

Asilomar 2025

[Paper]

Satellite Selection for In-Band Coexistence of Dense LEO Networks

Eunsun Kim, Ian P. Roberts, Taekyun Lee, Jeffrey G. Andrews

IEEE Transactions on Wireless Communications

[Paper]

Generating Informative Samples for Risk-Averse Fine-Tuning of Downstream Tasks

Heasung Kim, Taekyun Lee, Hyeji Kim, Gustavo de Veciana

NeurIPS 2025 Spotlight

[Project Page] [Paper] [OpenReview]

DiffLoc: Diffusion Model-Based High-Precision Positioning for 6G Networks

Taekyun Lee, Tommaso Balercia, Heasung Kim, Hyeji Kim, Jeffrey G. Andrews

ArXiv 2025

[Paper]

Generating High Dimensional User-Specific Wireless Channels using Diffusion Models

Taekyun Lee, Juseong Park, Hyeji Kim, Jeffrey G. Andrews

IEEE Transactions on Wireless Communications

[Paper] [Code]

Generative Diffusion Model-based Compression of MIMO CSI

Heasung Kim, Taekyun Lee, Hyeji Kim, Gustavo de Veciana, Mohamed Amine Arfaoui, Asil Koc, Phil Pietraski, Guodong Zhang, John Kaewell

ICC 2025

[Paper]

Importance Sampling in Digital Twin Networks

Taekyun Lee, Hyeji Kim, Jeffrey G. Andrews

IEEE CTW 2025

Self-Supervised Graph Contrastive Pretraining for Device-level Integrated Circuits

Sungyoung Lee, Ziyi Wang, Seunggeun Kim, Taekyun Lee, Yao Lai, David Z. Pan

ArXiv 2025

[Paper]

High Dimensional User-Specific Channel Generation for CSI Compression using Diffusion Models

Taekyun Lee, Juseong Park, Hyeji Kim, Jeffrey G. Andrews

Asilomar 2024

[Paper]

Deep Learning-Based mmWave Beam Alignment with Only Pilot Channel Measurements

Taekyun Lee, Hyeji Kim, Jeffrey G. Andrews

ICC 2024

[Paper]