← Search

Yu Rong

56 accepted papers

2026

DePO: Demonstration-guided Policy Optimization for Molecular Optimization

ICLR 2026poster

Large language models (LLMs) exhibit remarkable mathematical reasoning abilities through supervised fine-tuning (SFT) or reinforcement learning with verifiable rewards (RLVR). However, adapting them to scientific domains like molecular optimization is challenging: its datasets provide only reference…

Cited by 0SourceScholar
2026

Flow Along the $K$-Amplitude for Generative Modeling

ICLR 2026poster

In this work, we propose K-Flow, a novel generative learning paradigm that flows along the $K$-amplitude domain, where $K$ is a scaling parameter that organizes projected coefficients (frequency bands), and amplitude refers to the norm of such coefficients. We instantiate K-Flow with three concrete…

Cited by 0SourcecodeScholar
2026

Intervene When It Doubts: Conjunction-Guided Interactive Reasoning

ICML 2026poster

Large Reasoning Models (LRMs) excel at complex reasoning but suffer from inefficient reasoning, like overthinking and overshoot. These issues stem from excessive or misdirected reasoning triggered by the model's "doubt", manifested as self-validation and exploratory extension, increasing computation…

Cited by 0SourceScholar
2026

Large-scale Codec Avatars: The Unreasonable Effectiveness of Large-scale Avatar Pretraining

CVPR 2026

High-quality 3D avatar modeling faces a critical trade-off between fidelity and generalization. On the one hand, multi-view studio data enables high-fidelity modeling of humans with precise control over expressions and poses, but it struggles to generalize to real-world data due to limited scale and

Cited by 0SourcecodeScholar
2026

STAR-R1: Multi-View Spatial TrAnsformation Reasoning by Reinforcing Multimodal LLMs

CVPR 2026

Multimodal Large Language Models (MLLMs) remain far from human-level performance in multi-view spatial reasoning, where models must establish object correspondences across view and infer coherent scene semantics. We analyze this limitation through the Transformation-Driven Visual Reasoning (TVR) tas

Cited by 0SourcecodeScholar
2025

Analyzing LLMs’ Knowledge Boundary Cognition Across Languages Through the Lens of Internal Representations

ACL 2025long

While understanding the knowledge boundaries of LLMs is crucial to prevent hallucination, research on the knowledge boundaries of LLMs has predominantly focused on English. In this work, we present the first study to analyze how LLMs recognize knowledge boundaries across different languages by probi…

2025

Atomas: Hierarchical Adaptive Alignment on Molecule-Text for Unified Molecule Understanding and Generation

ICLR 2025poster

Molecule-and-text cross-modal representation learning has emerged as a promising direction for enhancing the quality of molecular representation, thereby improving performance in various scientific fields. However, most approaches employ a global alignment approach to learn the knowledge from differ…

2025

Chain of Ideas: Revolutionizing Research Via Novel Idea Development with LLM Agents

EMNLP 2025

Research ideation is crucial for scientific progress, but the exponential increase in scientific literature makes it challenging to stay updated and identify impactful directions. Recent developments in large language models(LLMs) offer a promising avenue to automate this process. However, existing

2025

CirT: Global Subseasonal-to-Seasonal Forecasting with Geometry-inspired Transformer

ICLR 2025poster

Accurate Subseasonal-to-Seasonal (S2S) climate forecasting is pivotal for decision-making including agriculture planning and disaster preparedness but is known to be challenging due to its chaotic nature. Although recent data-driven models have shown promising results, their performance is limited b…

2025

Disentangling Language and Culture for Evaluating Multilingual Large Language Models

ACL 2025long

This paper introduces a Dual Evaluation Framework to comprehensively assess the multilingual capabilities of LLMs. By decomposing the evaluation along the dimensions of linguistic medium and cultural context, this framework enables a nuanced analysis of LLMs’ ability to process questions within both…

Cited by 0SourcePDFScholar
2025

FineReason: Evaluating and Improving LLMs’ Deliberate Reasoning through Reflective Puzzle Solving

ACL 2025long

Many challenging reasoning tasks require not just rapid, intuitive responses, but a more deliberate, multi-step approach. Recent progress in large language models (LLMs) highlights an important shift from the “System 1” way of quick reactions to the “System 2” style of reflection-and-correction prob…

2025

GeoPQA: Bridging the Visual Perception Gap in MLLMs for Geometric Reasoning

EMNLP 2025

Recent advancements in reinforcement learning (RL) have enhanced the reasoning abilities of large language models (LLMs), yet the impact on multimodal LLMs (MLLMs) is limited. Particularly in vision-intensive tasks like geometric reasoning, MLLMs hallucinate frequently, leading to inaccurate reasoni

2025

IBCircuit: Towards Holistic Circuit Discovery with Information Bottleneck

ICML 2025poster

Circuit discovery has recently attracted attention as a potential research direction to explain the non-trivial behaviors of language models. It aims to find the computational subgraphs, also known as circuits, within the model that are responsible for solving specific tasks. However, most existing…

Cited by 0SourcePDFScholar
2025

InversionGNN: A Dual Path Network for Multi-Property Molecular Optimization

ICLR 2025poster

Exploring chemical space to find novel molecules that simultaneously satisfy multiple properties is crucial in drug discovery. However, existing methods often struggle with trading off multiple properties due to the conflicting or correlated nature of chemical properties. To tackle this issue, we i…

2025

LUCAS: Layered Universal Codec Avatars

CVPR 2025poster

Photorealistic 3D head avatar reconstruction faces critical challenges in modeling dynamic face-hair interactions and achieving cross-identity generalization, particularly during expressions and head movements. We present LUCAS, a novel Universal Prior Model (UPM) for codec avatar modeling that dise…

Cited by 0SourcePDFScholar
2025

Large Language-Geometry Model: When LLM meets Equivariance

ICML 2025poster

Accurately predicting 3D structures and dynamics of physical systems is crucial in scientific applications. Existing approaches that rely on geometric Graph Neural Networks (GNNs) effectively enforce $\mathrm{E}(3)$-equivariance, but they often fail in leveraging extensive broader information. While…

Cited by 4SourcePDFScholar
2025

MOF-BFN: Metal-Organic Frameworks Structure Prediction via Bayesian Flow Networks

NeurIPS 2025poster

Metal-Organic Frameworks (MOFs) have attracted considerable attention due to their unique properties including high surface area and tunable porosity, and promising applications in catalysis, gas storage, and drug delivery. Structure prediction for MOFs is a challenging task, as these frameworks are…

Cited by 0SourceScholar
2025

MolSpectra: Pre-training 3D Molecular Representation with Multi-modal Energy Spectra

ICLR 2025poster

Establishing the relationship between 3D structures and the energy states of molecular systems has proven to be a promising approach for learning 3D molecular representations. However, existing methods are limited to modeling the molecular energy states from classical mechanics. This limitation resu…

2025

Non-stationary Equivariant Graph Neural Networks for Physical Dynamics Simulation

NeurIPS 2025poster

To enhance the generalization ability of graph neural networks (GNNs) in learning and simulation physical dynamics, a series of equivariant GNNs have been developed to incorporate the symmetric inductive bias. However, the existing methods do not take into account the non-stationarity nature of phys…

Cited by 0SourcecodeScholar
2025

ReasonMed: A 370K Multi-Agent Generated Dataset for Advancing Medical Reasoning

EMNLP 2025

Reasoning-based large language models have excelled in mathematics and programming, yet their potential in knowledge-intensive medical question answering remains underexplored and insufficiently validated in clinical contexts. To bridge this gap, we introduce ReasonMed , the largest medical reasonin

2025

Scaling Language-centric Omnimodal Representation Learning

NeurIPS 2025poster

Recent multimodal embedding approaches leveraging multimodal large language models (MLLMs) fine-tuned with contrastive learning (CL) have shown promising results, yet the underlying reasons behind their superiority remain underexplored. This work argues that a crucial advantage of MLLM-based approac…

Cited by 0SourcecodeScholar
2025

The Rise of Parameter Specialization for Knowledge Storage in Large Language Models

NeurIPS 2025poster

Over time, a growing wave of large language models from various series has been introduced to the community. Researchers are striving to maximize the performance of language models with constrained parameter sizes. However, from a microscopic perspective, there has been limited research on how to be…

Cited by 0SourceScholar
2025

Universally Invariant Learning in Equivariant GNNs

NeurIPS 2025poster

Equivariant Graph Neural Networks (GNNs) have demonstrated significant success across various applications. To achieve completeness---that is, the universal approximation property over the space of equivariant functions---the network must effectively capture the intricate multi-body interactions amo…

Cited by 0SourceScholar
2024

Neural Atoms: Propagating Long-range Interaction in Molecular Graphs through Efficient Communication Channel

ICLR 2024poster

Graph Neural Networks (GNNs) have been widely adopted for drug discovery with molecular graphs. Nevertheless, current GNNs mainly excel in leveraging short-range interactions (SRI) but struggle to capture long-range interactions (LRI), both of which are crucial for determining molecular properties.…

2024

SEGNO: Generalizing Equivariant Graph Neural Networks with Physical Inductive Biases

ICLR 2024spotlight

Graph Neural Networks (GNNs) with equivariant properties have emerged as powerful tools for modeling complex dynamics of multi-object physical systems. However, their generalization ability is limited by the inadequate consideration of physical inductive biases: (1) Existing studies overlook the con…

Cited by 22SourcePDFScholar
2023

Be Selfish, But Wisely: Investigating the Impact of Agent Personality in Mixed-Motive Human-Agent Interactions

EMNLP 2023long main

A natural way to design a negotiation dialogue system is via self-play RL: train an agent that learns to maximize its performance by interacting with a simulated user that has been designed to imitate human-human dialogue data. Although this procedure has been adopted in prior work, we find that it…

Cited by 0SourceScholar
2023

Deep Insights into Noisy Pseudo Labeling on Graph Data

NeurIPS 2023poster

Pseudo labeling (PL) is a wide-applied strategy to enlarge the labeled dataset by self-annotating the potential samples during the training process. Several works have shown that it can improve the graph learning model performance in general. However, we notice that the incorrect labels can be fatal…

2023

DrugOOD: Out-of-Distribution Dataset Curator and Benchmark for AI-Aided Drug Discovery – a Focus on Affinity Prediction Problems with Noise Annotations

AAAI 2023technical

AI-aided drug discovery (AIDD) is gaining popularity due to its potential to make the search for new pharmaceuticals faster, less expensive, and more effective. Despite its extensive use in numerous fields (e.g., ADMET prediction, virtual screening), little research has been conducted on the out-of-…

Cited by 122SourcePDFScholar
2023

Energy-Motivated Equivariant Pretraining for 3D Molecular Graphs

AAAI 2023technical

Pretraining molecular representation models without labels is fundamental to various applications. Conventional methods mainly process 2D molecular graphs and focus solely on 2D tasks, making their pretrained models incapable of characterizing 3D geometry and thus defective for downstream 3D tasks.…

2023

Equivariant Spatio-Temporal Attentive Graph Networks to Simulate Physical Dynamics

NeurIPS 2023poster

Learning to represent and simulate the dynamics of physical systems is a crucial yet challenging task. Existing equivariant Graph Neural Network (GNN) based methods have encapsulated the symmetry of physics, \emph{e.g.}, translations, rotations, etc, leading to better generalization ability. Neverth…

2023

Human Mobility Modeling during the COVID-19 Pandemic via Deep Graph Diffusion Infomax

AAAI 2023technical

Non-Pharmaceutical Interventions (NPIs), such as social gathering restrictions, have shown effectiveness to slow the transmission of COVID-19 by reducing the contact of people. To support policy-makers, multiple studies have first modelled human mobility via macro indicators (e.g., average daily tra…

2023

Wireless Sensing for Simultaneous Human Vocal Sound and Heart Sound Recognition

ICASSP 2023accepted

Remote vibrometry using wireless signals is a recently introduced novel technique with a wide range of applications such as remote microphones and structural health monitoring. These use cases require high sensitivity and coherence in the sensing system. In this context, radar is a suitable all-weat…

Cited by 4SourceScholar
2022

Energy-Based Learning for Cooperative Games, with Applications to Valuation Problems in Machine Learning

ICLR 2022poster

Valuation problems, such as feature interpretation, data valuation and model valuation for ensembles, become increasingly more important in many machine learning applications. Such problems are commonly solved by well-known game-theoretic criteria, such as Shapley value or Banzhaf value. In this wor…

Cited by 18SourcePDFScholar
2022

Equivariant Graph Mechanics Networks with Constraints

ICLR 2022poster

Learning to reason about relations and dynamics over multiple interacting objects is a challenging topic in machine learning. The challenges mainly stem from that the interacting systems are exponentially-compositional, symmetrical, and commonly geometrically-constrained. Current methods, particular…

2022

Fine-Tuning Graph Neural Networks via Graph Topology Induced Optimal Transport

IJCAI 2022poster

Recently, the pretrain-finetuning paradigm has attracted tons of attention in graph learning community due to its power of alleviating the lack of labels problem in many real-world applications. Current studies use existing techniques, such as weight constraint, representation constraint, which are…

2022

Local Augmentation for Graph Neural Networks

ICML 2022spotlight

Graph Neural Networks (GNNs) have achieved remarkable performance on graph-based tasks. The key idea for GNNs is to obtain informative representation through aggregating information from local neighborhoods. However, it remains an open question whether the neighborhood information is adequately aggr…

2022

Towards Diverse and Natural Scene-Aware 3D Human Motion Synthesis

CVPR 2022poster

The ability to synthesize long-term human motion sequences in real-world scenes can facilitate numerous applications. Previous approaches for scene-aware motion synthesis are constrained by pre-defined target objects or positions and thus limit the diversity of human-scene interactions for synthesiz…

Cited by 90PDFScholar
2021

Exploring Robustness of Unsupervised Domain Adaptation in Semantic Segmentation

ICCV 2021poster

Recent studies imply that deep neural networks are vulnerable to adversarial examples, i.e., inputs with a slight but intentional perturbation are incorrectly classified by the network. Such vulnerability makes it risky for some security-related applications (e.g., semantic segmentation in autonomou…

Cited by 45PDFcodeScholar
2021

Graph Information Bottleneck for Subgraph Recognition

ICLR 2021poster

Given the input graph and its label/property, several key problems of graph learning, such as finding interpretable subgraphs, graph denoising and graph compression, can be attributed to the fundamental problem of recognizing a subgraph of the original one. This subgraph shall be as informative…

2021

Hierarchical Graph Capsule Network

AAAI 2021technical

Graph Neural Networks (GNNs) draw their strength from explicitly modeling the topological information of structured data. However, existing GNNs suffer from limited capability in capturing the hierarchical graph representation which plays an important role in graph classification. In this paper, we…

2021

Learning Diverse-Structured Networks for Adversarial Robustness

ICML 2021spotlight

In adversarial training (AT), the main focus has been the objective and optimizer while the model has been less studied, so that the models being used are still those classic ones in standard training (ST). Classic network architectures (NAs) are generally worse than searched NA in ST, which should…

2021

Not All Low-Pass Filters are Robust in Graph Convolutional Networks

NeurIPS 2021poster

Graph Convolutional Networks (GCNs) are promising deep learning approaches in learning representations for graph-structured data. Despite the proliferation of such methods, it is well known that they are vulnerable to carefully crafted adversarial attacks on the graph structure. In this paper, we fi…

2020

Deep Multimodal Fusion by Channel Exchanging

NeurIPS 2020poster

Deep multimodal fusion by using multiple sources of data for classification or regression has exhibited a clear advantage over the unimodal counterpart on various applications. Yet, current methods including aggregation-based and alignment-based fusion are still inadequate in balancing the trade-off…

2020

Dirichlet Graph Variational Autoencoder

NeurIPS 2020poster

Graph Neural Networks (GNN) and Variational Autoencoders (VAEs) have been widely used in modeling and generating graphs with latent factors. However there is no clear explanation of what these latent factors are and why they perform well. In this work, we present Dirichlet Graph Variational Autoenco…

Cited by 55SourcePDFScholar
2020

DropEdge: Towards Deep Graph Convolutional Networks on Node Classification

ICLR 2020poster

Over-fitting and over-smoothing are two main obstacles of developing deep Graph Convolutional Networks (GCNs) for node classification. In particular, over-fitting weakens the generalization ability on small dataset, while over-smoothing impedes model training by isolating output representations from…

Cited by 1783SourcecodeScholar
2020

Self-Supervised Graph Transformer on Large-Scale Molecular Data

NeurIPS 2020poster

How to obtain informative representations of molecules is a crucial prerequisite in AI-driven drug design and discovery. Recent researches abstract molecules as graphs and employ Graph Neural Networks (GNNs) for molecular representation learning. Nevertheless, two issues impede the usage of GNNs in…

2019

Delving Deep Into Hybrid Annotations for 3D Human Recovery in the Wild

ICCV 2019poster

Though much progress has been achieved in single-image 3D human recovery, estimating 3D model for in-the-wild images remains a formidable challenge. The reason lies in the fact that obtaining high-quality 3D annotations for in-the-wild images is an extremely hard task that consumes enormous amount o…

Cited by 73PDFScholar
2019

Graph Convolutional Networks for Temporal Action Localization

ICCV 2019poster

Most state-of-the-art action localization systems process each action proposal individually, without explicitly exploiting their relations during learning. However, the relations between proposals actually play an important role in action localization, since a meaningful action always consists of mu…

Cited by 640PDFcodeScholar
2019

Progressive Feature Alignment for Unsupervised Domain Adaptation

CVPR 2019poster

Unsupervised domain adaptation (UDA) transfers knowledge from a label-rich source domain to a fully-unlabeled target domain. To tackle this task, recent approaches resort to discriminative domain transfer in virtue of pseudo-labels to enforce the class-level distribution alignment across the source…

Cited by 544PDFScholar
2018

Adaptive Sampling Towards Fast Graph Representation Learning

NeurIPS 2018poster

Graph Convolutional Networks (GCNs) have become a crucial tool on learning representations of graph vertices. The main challenge of adapting GCNs on large-scale graphs is the scalability issue that it incurs heavy cost both in computation and memory due to the uncontrollable neighborhood expansion a…

Cited by 640SourcePDFScholar
2018

Pose-Robust Face Recognition via Deep Residual Equivariant Mapping

CVPR 2018poster

Face recognition achieves exceptional success thanks to the emergence of deep learning. However, many contemporary face recognition models still perform relatively poor in processing profile faces compared to frontal faces. A key reason is that the number of frontal and profile training faces are hi…

Cited by 185SourcePDFScholar