← Search

Joohwan Ko

6 accepted papers

2026

ConvT3: Structured State Kernels for Convolutional State Space Models

ICLR 2026poster

Modeling long spatiotemporal sequences requires capturing both complex spatial correlations and temporal dependencies. Convolutional State Space Models (ConvSSMs) have been proposed to incorporate spatial modeling in State Space Models (SSMs) using the convolution of tensor-valued states and kernels…

Cited by 0SourceScholar
2024

Layer-Adaptive State Pruning for Deep State Space Models

NeurIPS 2024poster

Due to the lack of state dimension optimization methods, deep state space models (SSMs) have sacrificed model capacity, training search space, or stability to alleviate computational costs caused by high state dimensions. In this work, we provide a structured pruning method for SSMs, Layer-Adaptive…

2024

Learning to Scale Logits for Temperature-Conditional GFlowNets

ICML 2024poster

GFlowNets are probabilistic models that sequentially generate compositional structures through a stochastic policy. Among GFlowNets, temperature-conditional GFlowNets can introduce temperature-based controllability for exploration and exploitation. We propose *Logit-scaling GFlowNets* (Logit-GFN), a…

2024

Provably Scalable Black-Box Variational Inference with Structured Variational Families

ICML 2024poster

Variational families with full-rank covariance approximations are known not to work well in black-box variational inference (BBVI), both empirically and theoretically. In fact, recent computational complexity results for BBVI have established that full-rank variational families scale poorly with the…

Cited by 0SourcePDFScholar