← Search

Hengrui Zhang

20 accepted papers

2026

Bee: A High-Quality Corpus and Full-Stack Suite to Unlock Advanced Fully Open MLLMs

ICLR 2026poster

Fully open multimodal large language models (MLLMs) currently lag behind proprietary counterparts, primarily due to a significant gap in data quality for supervised fine-tuning (SFT). Existing open-source datasets are often plagued by widespread noise and a critical deficit in complex reasoning dat…

Cited by 0SourceScholar
2026

MUSE: Model-Agnostic Tabular Watermarking via Multi-Sample Selection

ICLR 2026poster

We introduce MUSE, a novel watermarking paradigm for tabular generative models. Existing approaches often exploit DDIM invertibility to watermark tabular diffusion models, but tabular diffusion models suffer from poor invertibility, leading to degraded performance. To overcome this limitation, we le…

Cited by 0SourcecodeScholar
2026

ROAD: Responsibility-Oriented Reward Design for Reinforcement Learning in Autonomous Driving

RA-L 2026

Reinforcement learning (RL) in autonomous driving employs a trial-and-error mechanism, enhancing robustness in unpredictable environments. However, crafting effective reward functions remains challenging, as conventional approaches rely heavily on manual design and demonstrate limited efficacy in co

Cited by 0SourceScholar
2026

Return-Critic: Bridging Goal Discrepancy for Efficient Visual Reinforcement Learning

ICML 2026poster

Sample inefficiency remains a challenge in pixel-based visual reinforcement learning (RL), primarily due to ineffective state representation learning. While recent advances employ auxiliary tasks to improve representation learning, their representation goals (e.g., mask reconstruction, state predict…

Cited by 0SourceScholar
2025

DiffPuter: Empowering Diffusion Models for Missing Data Imputation

ICLR 2025spotlight

Generative models play an important role in missing data imputation in that they aim to learn the joint distribution of full data. However, applying advanced deep generative models (such as Diffusion models) to missing data imputation is challenging due to 1) the inherent incompleteness of the train…

Cited by 2SourcePDFScholar
2025

PathwiseRAG: Multi-Dimensional Exploration and Integration Framework

EMNLP 2025

Conventional retrieval-augmented generation(RAG) systems employ rigid retrieval strategies that create: (1) knowledge blind spots across domain boundaries, (2) reasoning fragmentation when processing interdependent concepts, and (3) contradictions from conflicting evidence sources. Motivated by thes

Cited by 0SourcePDFScholar
2025

TABGEN-ICL: Residual-Aware In-Context Example Selection for Tabular Data Generation

ACL 2025finding

Large Language models (LLMs) have achieved encouraging results in tabular data generation. However, existing approaches require fine-tuning, which is computationally expensive. This paper explores an alternative: prompting a fixed LLM with in-context examples. We observe that using randomly selected…

2025

TabDiff: a Mixed-type Diffusion Model for Tabular Data Generation

ICLR 2025poster

Synthesizing high-quality tabular data is an important topic in many data science tasks, ranging from dataset augmentation to privacy protection. However, developing expressive generative models for tabular data is challenging due to its inherent heterogeneous data types, complex inter-correlations,…

2025

TabNAT: A Continuous-Discrete Joint Generative Framework for Tabular Data

ICML 2025poster

While autoregressive models dominate natural language generation, their application to tabular data remains limited due to two challenges: 1) tabular data contains heterogeneous types, whereas autoregressive next-token (distribution) prediction is designed for discrete data, and 2) tabular data is c…

Cited by 0SourcePDFScholar
2024

Enhancing Off-Policy Constrained Reinforcement Learning through Adaptive Ensemble C Estimation

AAAI 2024technical

In the domain of real-world agents, the application of Reinforcement Learning (RL) remains challenging due to the necessity for safety constraints. Previously, Constrained Reinforcement Learning (CRL) has predominantly focused on on-policy algorithms. Although these algorithms exhibit a degree of ef…

Cited by 1SourcePDFScholar
2024

Exploitation of a Latent Mechanism in Graph Contrastive Learning: Representation Scattering

NeurIPS 2024oral

Graph Contrastive Learning (GCL) has emerged as a powerful approach for generating graph representations without the need for manual annotation. Most advanced GCL methods fall into three main frameworks: node discrimination, group discrimination, and bootstrapping schemes, all of which achieve compa…

2024

Kraken: Inherently Parallel Transformers For Efficient Multi-Device Inference

NeurIPS 2024poster

Large Transformer networks are increasingly used in settings where low inference latency is necessary to enable new applications and improve the end-user experience. However, autoregressive inference is resource intensive and requires parallelism for efficiency. Parallelism introduces collective com…

Cited by 2SourcePDFScholar
2024

Mixed-Type Tabular Data Synthesis with Score-based Diffusion in Latent Space

ICLR 2024oral

Recent advances in tabular data generation have greatly enhanced synthetic data quality. However, extending diffusion models to tabular data is challenging due to the intricately varied distributions and a blend of data types of tabular data. This paper introduces TabSyn, a methodology that synthesi…

2023

SGFormer: Simplifying and Empowering Transformers for Large-Graph Representations

NeurIPS 2023poster

Learning representations on large-sized graphs is a long-standing challenge due to the inter-dependence nature involved in massive data points. Transformers, as an emerging class of foundation encoders for graph-structured data, have shown promising performance on small graphs due to its global atte…

2022

Align Representations With Base: A New Approach to Self-Supervised Learning

CVPR 2022poster

Existing symmetric contrastive learning methods suffer from collapses (complete and dimensional) or quadratic complexity of objectives. Departure from these methods which maximize mutual information of two generated views, along either instance or feature dimension, the proposed paradigm introduces…

Cited by 30PDFScholar
2022

Handling Distribution Shifts on Graphs: An Invariance Perspective

ICLR 2022poster

There is increasing evidence suggesting neural networks' sensitivity to distribution shifts, so that research on out-of-distribution (OOD) generalization comes into the spotlight. Nonetheless, current endeavors mostly focus on Euclidean data, and its formulation for graph-structured data is not clea…

2021

From Canonical Correlation Analysis to Self-supervised Graph Neural Networks

NeurIPS 2021poster

We introduce a conceptually simple yet effective model for self-supervised representation learning with graph data. It follows the previous methods that generate two views of an input graph through data augmentation. However, unlike contrastive methods that focus on instance-level discrimination, we…

2021

Super Odometry: IMU-centric LiDAR-Visual-Inertial Estimator for Challenging Environments

IROS 2021poster

We propose Super Odometry, a high-precision multi-modal sensor fusion framework, providing a simple but effective way to fuse multiple sensors such as LiDAR, camera, and IMU sensors and achieve robust state estimation in perceptually-degraded environments. Different from traditional sensor-fusion me…

Cited by 203SourceScholar
2021

Towards Open-World Recommendation: An Inductive Model-based Collaborative Filtering Approach

ICML 2021spotlight

Recommendation models can effectively estimate underlying user interests and predict one’s future behaviors by factorizing an observed user-item rating matrix into products of two sets of latent factors. However, the user-specific embedding factors can only be learned in a transductive way, making i…

2020

TP-TIO: A Robust Thermal-Inertial Odometry with Deep ThermalPoint

IROS 2020poster

To achieve robust motion estimation in visually degraded environments, thermal odometry has been an attraction in the robotics community. However, most thermal odometry methods are purely based on classical feature extractors, which is difficult to establish robust correspondences in successive fram…

Cited by 49SourceScholar