← Search

Kookjin Lee

19 accepted papers

2026

Adversarial Robustness of Implicit Neural Representation-Based Classifiers

ICML 2026poster

Implicit neural representations (INRs) encode data as continuous coordinate-based functions parameterized by neural networks, shifting downstream tasks such as image recognition to operate on functional rather than discrete representations. Despite their increasing adoption, the adversarial robustne…

Cited by 0SourceScholar
2026

Basis-Oriented Low-rank Transfer for Few-Shot and Test-Time Adaptation

CVPR 2026

Adapting large pre-trained models to unseen tasks under tight data and compute budgets remains challenging. Meta-learning approaches explicitly learn good initializations, but they require an additional meta-training phase over many tasks, incur high training cost, and can be unstable. At the same t

Cited by 0SourceScholar
2026

Extending Fourier Neural Operators for Modeling Parameterized and Coupled PDEs

ICLR 2026poster

Parameterized and coupled partial differential equations (PDEs) are central to modeling phenomena in science and engineering, yet neural operator methods that address both aspects remain limited. We extend Fourier neural operators (FNOs) with minimal architectural modifications along two directions.…

Cited by 0SourceScholar
2026

Meta-learning Structure-Preserving Dynamics

ICML 2026poster

Structure-preserving approaches to dynamics discovery have demonstrated great potential for modeling physical systems due to their use of strong inductive biases, which enforce key features such as conservation laws and dissipative behavior. However, these models are typically trained on a per-confi…

Cited by 1SourceScholar
2025

Efficiently Parameterized Neural Metriplectic Systems

ICLR 2025poster

Metriplectic systems are learned from data in a way that scales quadratically in both the size of the state and the rank of the metriplectic operators. In addition to being provably energy-conserving and entropy-stable, the proposed neural metriplectic systems (NMS) approach includes approximation…

Cited by 3SourcePDFScholar
2025

PDEfuncta: Spectrally-Aware Neural Representation for PDE Solution Modeling

NeurIPS 2025poster

Scientific machine learning often involves representing complex solution fields that exhibit high-frequency features such as sharp transitions, fine-scale oscillations, and localized structures. While implicit neural representations (INRs) have shown promise for continuous function modeling, capturi…

Cited by 0SourceScholar
2025

PIORF: Physics-Informed Ollivier-Ricci Flow for Long–Range Interactions in Mesh Graph Neural Networks

ICLR 2025poster

Recently, data-driven simulators based on graph neural networks have gained attention in modeling physical systems on unstructured meshes. However, they struggle with long-range dependencies in fluid flows, particularly in refined mesh regions. This challenge, known as the 'over-squashing' problem,…

Cited by 5SourcePDFScholar
2024

Graph Convolutions Enrich the Self-Attention in Transformers!

NeurIPS 2024poster

Transformers, renowned for their self-attention mechanism, have achieved state-of-the-art performance across various tasks in natural language processing, computer vision, time-series modeling, etc. However, one of the challenges with deep Transformer models is the oversmoothing problem, where repre…

2024

Learning Flexible Body Collision Dynamics with Hierarchical Contact Mesh Transformer

ICLR 2024poster

Recently, many mesh-based graph neural network (GNN) models have been proposed for modeling complex high-dimensional physical systems. Remarkable achievements have been made in significantly reducing the solving time compared to traditional numerical solvers. These methods are typically designed to…

2024

Operator-Learning-Inspired Modeling of Neural Ordinary Differential Equations

AAAI 2024technical

Neural ordinary differential equations (NODEs), one of the most influential works of the differential equation-based deep learning, are to continuously generalize residual networks and opened a new field. They are currently utilized for various downstream tasks, e.g., image classification, time seri…

Cited by 3SourcePDFScholar
2024

PAC-FNO: Parallel-Structured All-Component Fourier Neural Operators for Recognizing Low-Quality Images

ICLR 2024poster

A standard practice in developing image recognition models is to train a model on a specific image resolution and then deploy it. However, in real-world inference, models often encounter images different from the training sets in resolution and/or subject to natural variations such as weather change…

Cited by 0SourcePDFScholar
2024

Parameterized Physics-informed Neural Networks for Parameterized PDEs

ICML 2024oral

Complex physical systems are often described by partial differential equations (PDEs) that depend on parameters such as the Raynolds number in fluid mechanics. In applications such as design optimization or uncertainty quantification, solutions of those PDEs need to be evaluated at numerous points i…

Cited by 23SourcePDFScholar
2023

Hypernetwork-based Meta-Learning for Low-Rank Physics-Informed Neural Networks

NeurIPS 2023spotlight

In various engineering and applied science applications, repetitive numerical simulations of partial differential equations (PDEs) for varying input parameters are often required (e.g., aircraft shape optimization over many design parameters) and solvers are required to perform rapid execution. In t…

Cited by 26SourcePDFScholar
2023

Reversible and irreversible bracket-based dynamics for deep graph neural networks

NeurIPS 2023poster

Recent works have shown that physics-inspired architectures allow the training of deep graph neural networks (GNNs) without oversmoothing. The role of these physics is unclear, however, with successful examples of both reversible (e.g., Hamiltonian) and irreversible (e.g., diffusion) phenomena produ…

2021

A Novel Method to Solve Neural Knapsack Problems

ICML 2021spotlight

0-1 knapsack is of fundamental importance across many fields. In this paper, we present a game-theoretic method to solve 0-1 knapsack problems (KPs) where the number of items (products) is large and the values of items are not predetermined but decided by an external value assignment function (e.g.,…

Cited by 10SourcePDFScholar
2021

DPM: A Novel Training Method for Physics-Informed Neural Networks in Extrapolation

AAAI 2021technical

We present a method for learning dynamics of complex physical processes described by time-dependent nonlinear partial differential equations (PDEs). Our particular interest lies in extrapolating solutions in time beyond the range of temporal domain used in training. Our choice for a baseline method…

2021

Deep Conservation: A Latent-Dynamics Model for Exact Satisfaction of Physical Conservation Laws

AAAI 2021technical

This work proposes an approach for latent-dynamics learning that exactly enforces physical conservation laws. The method comprises two steps. First, the method computes a low-dimensional embedding of the high-dimensional dynamical-system state using deep convolutional autoencoders. This defines a lo…

Cited by 66SourcePDFScholar
2021

Machine learning structure preserving brackets for forecasting irreversible processes

NeurIPS 2021poster

Forecasting of time-series data requires imposition of inductive biases to obtain predictive extrapolation, and recent works have imposed Hamiltonian/Lagrangian form to preserve structure for systems with \emph{reversible} dynamics. In this work we present a novel parameterization of dissipative bra…

Cited by 53SourcePDFScholar