papers

Publications (18)

cs.CV2025

BrainMorph: A Foundational Keypoint Model for Robust and Flexible Brain MRI Registration

Alan Q. Wang, Rachit Saluja, Heejong Kim +3

We present a keypoint-based foundation model for general purpose brain MRI registration, based on the recently-proposed KeyMorph framework. Our model, called BrainMorph, serves as…

cs.LG2024

Adapting to Shifting Correlations with Unlabeled Data Calibration

Minh Nguyen, Alan Q. Wang, Heejong Kim +1

Distribution shifts between sites can seriously degrade model performance since models are prone to exploiting unstable correlations. Thus, many methods try to find features that a…

cs.LG2026

MAdam: Metric-Aware Multi-Objective Adam

Fengbei Liu, Rachit Saluja, Sunwoo Kwak +5

Multi-objective optimization (MOO) underlies many machine learning problems, yet MOO solvers across the loss-balancing, gradient-balancing, and Pareto-based families almost univers…

eess.IV2023

Learning to Compare Longitudinal Images

Heejong Kim, Mert R. Sabuncu

Longitudinal studies, where a series of images from the same set of individuals are acquired at different time-points, represent a popular technique for studying and characterizing…

eess.IV2020

A framework to construct a longitudinal DW-MRI infant atlas based on mixed effects modeling of dODF coefficients

Heejong Kim, Martin Styner, Joseph Piven +1

Building of atlases plays a crucial role in the analysis of brain images. In scenarios where early growth, aging or disease trajectories are of key importance, longitudinal atlases…

eess.IV2024

Effective Segmentation of Post-Treatment Gliomas Using Simple Approaches: Artificial Sequence Generation and Ensemble Models

Heejong Kim, Leo Milecki, Mina C Moghadam +5

Segmentation is a crucial task in the medical imaging field and is often an important primary step or even a prerequisite to the analysis of medical volumes. Yet treatments such as…