← Search

Shih-Hsin Wang

7 accepted papers

2026

RMFlow: Refined Mean Flow by a Noise-Injection Step for Multimodal Generation

ICLR 2026poster

Mean flow (MeanFlow) enables efficient, high-fidelity image generation, yet its single-function evaluation (1-NFE) generation often cannot yield compelling results. We address this issue by introducing RMFlow, an efficient multimodal generative model that integrates a coarse 1-NFE MeanFlow transport…

Cited by 0SourceScholar
2026

Test-Time Guidance for Flow-Based Generative Models via Parallel Tempering on Source Distributions

ICML 2026poster

Generative models that transport a simple source distribution to a complex data distribution—such as diffusion and flow-based models—are central to high‑fidelity data generation. Test-time guidance can further steer pretrained models toward user-specified high-reward regions without costly retrainin…

Cited by 0SourceScholar
2025

A Theoretically-Principled Sparse, Connected, and Rigid Graph Representation of Molecules

ICLR 2025oral

Graph neural networks (GNNs) -- learn graph representations by exploiting the graph's sparsity, connectivity, and symmetries -- have become indispensable for learning geometric data like molecules. However, the most used graphs (e.g., radial cutoff graphs) in molecular modeling lack theoretical guar…

2025

Improving Flow Matching by Aligning Flow Divergence

ICML 2025poster

Conditional flow matching (CFM) stands out as an efficient, simulation-free approach for training flow-based generative models, achieving remarkable performance for data generation. However, CFM is insufficient to ensure accuracy in learning probability paths. In this paper, we introduce a new parti…

Cited by 0SourcePDFScholar
2025

Towards Multiscale Graph-based Protein Learning with Geometric Secondary Structural Motifs

NeurIPS 2025poster

Graph neural networks (GNNs) have emerged as powerful tools for learning protein structures by capturing spatial relationships at the residue level. However, existing GNN-based methods often face challenges in learning multiscale representations and modeling long-range dependencies efficiently. In t…

Cited by 0SourceScholar
2024

An Explicit Frame Construction for Normalizing 3D Point Clouds

ICML 2024poster

Many real-world datasets are represented as 3D point clouds -- yet they often lack a predefined reference frame, posing a challenge for machine learning or general data analysis. Traditional methods for determining reference frames and normalizing 3D point clouds often struggle with specific inputs,…

2024

Rethinking the Benefits of Steerable Features in 3D Equivariant Graph Neural Networks

ICLR 2024poster

Theoretical and empirical comparisons have been made to assess the expressive power and performance of invariant and equivariant GNNs. However, there is currently no theoretical result comparing the expressive power of $k$-hop invariant GNNs and equivariant GNNs. Additionally, little is understood a…

Cited by 7SourcePDFScholar