Publications (122)
DistFlow: A Fully Distributed RL Framework for Scalable and Efficient LLM Post-Training
Zhixin Wang, Jiaming Xu, Tianyi Zhou +10
Towards a Universal 3D Medical Multi-modality Generalization via Learning Personalized Invariant Representation
Zhaorui Tan, Xi Yang, Tan Pan +8
ResNorm: Tackling Long-tailed Degree Distribution Issue in Graph Neural Networks via Normalization
Langzhang Liang, Zenglin Xu, Zixing Song +3
Distributed Flexible Nonlinear Tensor Factorization
Shandian Zhe, Kai Zhang, Pengyuan Wang +4
Learning Implicit Bias in Generative Spaces for Accelerating Protein Dynamics Emulation
Kaihui Cheng, Zhiqiang Cai, Wenkai Xiang +4
Guiding Diffusion Models with Reinforcement Learning for Stable Molecule Generation
Zhijian Zhou, Junyi An, Zongkai Liu +5
InfMLLM: A Unified Framework for Visual-Language Tasks
Qiang Zhou, Zhibin Wang, Wei Chu +3
An Attentive Dual-Encoder Framework Leveraging Multimodal Visual and Semantic Information for Automatic OSAHS Diagnosis
Yingchen Wei, Xihe Qiu, Xiaoyu Tan +4
Conditional Attention Networks for Distilling Knowledge Graphs in Recommendation
Ke Tu, Peng Cui, Daixin Wang +4
The Choice of Divergence: A Neglected Key to Mitigating Diversity Collapse in Reinforcement Learning with Verifiable Reward
Long Li, Zhijian Zhou, Jiaran Hao +9
A Sensitivity Analysis of the Surrogate Index Approach for Estimating Long-Term Treatment Effects
Yanqin Fan, Carlos A. Manzanares, Hyeonseok Park +1
Riemannian Proximal Policy Optimization
Shijun Wang, Baocheng Zhu, Chen Li +4
Long Short-Term Sample Distillation
Liang Jiang, Zujie Wen, Zhongping Liang +7
Start Classifying: Categorical Critics for LLM Reinforcement Learning
Zhijian Zhou, Long Li, Xuan Zhang +7
ChromFound: Towards A Universal Foundation Model for Single-Cell Chromatin Accessibility Data
Yifeng Jiao, Yuchen Liu, Yu Zhang +9
Equivariant Spherical Transformer for Efficient Molecular Modeling
Junyi An, Xinyu Lu, Chao Qu +6
PDETime: Rethinking Long-Term Multivariate Time Series Forecasting from the perspective of partial differential equations
Shiyi Qi, Zenglin Xu, Yiduo Li +4
Reinforcement Learning for Uplift Modeling
Chenchen Li, Xiang Yan, Xiaotie Deng +6
PILLOW: Enhancing Efficient Instruction Fine-tuning via Prompt Matching
Zhenting Qi, Xiaoyu Tan, Shaojie Shi +3
ReTokSync: Self-Synchronizing Tokenization Disambiguation for Generative Linguistic Steganography
Yaofei Wang, Rui Wang, Weilong Pang +4
PAST: A multimodal single-cell foundation model for histopathology and spatial transcriptomics in cancer
Changchun Yang, Haoyang Li, Yushuai Wu +8
Value Propagation for Decentralized Networked Deep Multi-agent Reinforcement Learning
Chao Qu, Shie Mannor, Huan Xu +3
Enhancing the Reliability of Segment Anything Model for Auto-Prompting Medical Image Segmentation with Uncertainty Rectification
Yichi Zhang, Shiyao Hu, Sijie Ren +3
Suiren-1.0 Technical Report: A Family of Molecular Foundation Models
Junyi An, Xinyu Lu, Yun-Fei Shi +5
ChemHTS: Hierarchical Tool Stacking for Enhancing Chemical Agents
Zhucong Li, Jin Xiao, Bowei Zhang +5
FuXi-Extreme: Improving extreme rainfall and wind forecasts with diffusion model
Xiaohui Zhong, Lei Chen, Jun Liu +3
FuXi Weather: A data-to-forecast machine learning system for global weather
Xiuyu Sun, Xiaohui Zhong, Xiaoze Xu +8
How Much Can A Retailer Sell? Sales Forecasting on Tmall
Chaochao Chen, Ziqi Liu, Jun Zhou +4
SemiSAM: Enhancing Semi-Supervised Medical Image Segmentation via SAM-Assisted Consistency Regularization
Yichi Zhang, Jin Yang, Yuchen Liu +2
PET2Rep: Towards Vision-Language Model-Drived Automated Radiology Report Generation for Positron Emission Tomography
Yichi Zhang, Wenbo Zhang, Zehui Ling +12
FedCVD: The First Real-World Federated Learning Benchmark on Cardiovascular Disease Data
Yukun Zhang, Guanzhong Chen, Zenglin Xu +6
Latent Dirichlet Allocation for Internet Price War
Chenchen Li, Xiang Yan, Xiaotie Deng +6
Message passing with relaxed moment matching
Yuan Qi, Yandong Guo
FLAG: Foundation model representation with Latent diffusion Alignment via Graph for spatial gene expression prediction
Qi Si, Penglei Wang, Yushuai Wu +5
Equivariant Masked Position Prediction for Efficient Molecular Representation
Junyi An, Chao Qu, Yun-Fei Shi +4
Unleashing Flow Policies with Distributional Critics
Deshu Chen, Yuchen Liu, Zhijian Zhou +2
NetDP: An Industrial-Scale Distributed Network Representation Framework for Default Prediction in Ant Credit Pay
Jianbin Lin, Zhiqiang Zhang, Jun Zhou +5
Evaluation of Winning Solutions of 2025 Low Power Computer Vision Challenge
Zihao Ye, Yung-Hsiang Lu, Xiao Hu +14
InterveneBench: Benchmarking LLMs for Intervention Reasoning and Causal Study Design in Real Social Systems
Shaojie Shi, Zhengyu Shi, Lingran Zheng +15
Internalizing the Future: A Unified Agentic Training Paradigm for World Model Planning
Xuan Zhang, Zhijian Zhou, Lingfeng Qiao +6
Using Forests in Multivariate Regression Discontinuity Designs
Yiqi Liu, Yuan Qi
PET-F2I: A Comprehensive Benchmark and Parameter-Efficient Fine-Tuning of LLMs for PET/CT Report Impression Generation
Yuchen Liu, Wenbo Zhang, Liling Peng +6
Bandit Samplers for Training Graph Neural Networks
Ziqi Liu, Zhengwei Wu, Zhiqiang Zhang +4
AI-for-Science Low-code Platform with Bayesian Adversarial Multi-Agent Framework
Zihang Zeng, Jiaquan Zhang, Pengze Li +2
Thought-Like-Pro: Enhancing Reasoning of Large Language Models through Self-Driven Prolog-based Chain-of-Thought
Xiaoyu Tan, Yongxin Deng, Xihe Qiu +5
Structure-based RNA Design by Step-wise Optimization of Latent Diffusion Model
Qi Si, Xuyang Liu, Penglei Wang +3
A Semi-supervised Graph Attentive Network for Financial Fraud Detection
Daixin Wang, Jianbin Lin, Peng Cui +7
DyneTrion: A Spatio-temporally Coherent Generative Emulator for Protein Dynamics Across Timescales
Kaihui Cheng, Zhiqiang Cai, Peng Tu +6
Asynchronous Distributed Variational Gaussian Processes for Regression
Hao Peng, Shandian Zhe, Yuan Qi
Towards Collaborative Intelligence: Propagating Intentions and Reasoning for Multi-Agent Coordination with Large Language Models
Xihe Qiu, Haoyu Wang, Xiaoyu Tan +6
Supervised Heterogeneous Multiview Learning for Joint Association Study and Disease Diagnosis
Shandian Zhe, Zenglin Xu, Yuan Qi
Dynamic PDB: A New Dataset and a SE(3) Model Extension by Integrating Dynamic Behaviors and Physical Properties in Protein Structures
Ce Liu, Jun Wang, Zhiqiang Cai +10
The APC Algorithm of Solving Large-Scale Linear Systems: A Generalized Analysis
Jiyan Zhang, Yue Xue, Yuan Qi +1
SemiSAM-O1: How far can we push the boundary of annotation-efficient medical image segmentation?
Yichi Zhang, Le Xue, Bichun Xu +6
Efficient Probabilistic Logic Reasoning with Graph Neural Networks
Yuyu Zhang, Xinshi Chen, Yuan Yang +4
RNE: A Scalable Network Embedding for Billion-scale Recommendation
Jianbin Lin, Daixin Wang, Lu Guan +5
ChemAmp: Amplified Chemistry Tools via Composable Agents
Zhucong Li, Powei Chang, Jin Xiao +6
A Riemannian Primal-dual Algorithm Based on Proximal Operator and its Application in Metric Learning
Shijun Wang, Baocheng Zhu, Lintao Ma +1
Graph Representation Learning for Merchant Incentive Optimization in Mobile Payment Marketing
Ziqi Liu, Dong Wang, Qianyu Yu +8
Project and Generate: Divergence-Free Neural Operators for Incompressible Flows
Xigui Li, Hongwei Zhang, Ruoxi Jiang +6
Cost-Effective Incentive Allocation via Structured Counterfactual Inference
Romain Lopez, Chenchen Li, Xiang Yan +4
InfDetect: a Large Scale Graph-based Fraud Detection System for E-Commerce Insurance
Cen Chen, Chen Liang, Jianbin Lin +7
Incentive-Aware Recommender Systems in Two-Sided Markets
Xiaowu Dai, Wenlu Xu, Yuan Qi +1
Developing Foundation Models for Universal Segmentation from 3D Whole-Body Positron Emission Tomography
Yichi Zhang, Le Xue, Wenbo Zhang +16
OpenCoder: The Open Cookbook for Top-Tier Code Large Language Models
Siming Huang, Tianhao Cheng, J. K. Liu +16
CENTAUR: Bridging the Impossible Trinity of Privacy, Efficiency, and Performance in Privacy-Preserving Transformer Inference
Jinglong Luo, Guanzhong Chen, Yehong Zhang +6
SemiSAM+: Rethinking Semi-Supervised Medical Image Segmentation in the Era of Foundation Models
Yichi Zhang, Bohao Lv, Le Xue +5
Aneumo: A Large-Scale Comprehensive Synthetic Dataset of Aneurysm Hemodynamics
Xigui Li, Yuanye Zhou, Feiyang Xiao +10
Promoting Equality in Large Language Models: Identifying and Mitigating the Implicit Bias based on Bayesian Theory
Yongxin Deng, Xihe Qiu, Xiaoyu Tan +6
EigenNet: A Bayesian hybrid of generative and conditional models for sparse learning
Yuan Qi, Feng Yan
Modeling the Field Value Variations and Field Interactions Simultaneously for Fraud Detection
Dongbo Xi, Bowen Song, Fuzhen Zhuang +5
Efficient Network Automatic Relevance Determination
Hongwei Zhang, Ziqi Ye, Xinyuan Wang +5
AI2Apps: A Visual IDE for Building LLM-based AI Agent Applications
Xin Pang, Zhucong Li, Jiaxiang Chen +3
Harnessing Negative Signals: Reinforcement Distillation from Teacher Data for LLM Reasoning
Shuyao Xu, Cheng Peng, Jiangxuan Long +3
Count Counts: Motivating Exploration in LLM Reasoning with Count-based Intrinsic Rewards
Xuan Zhang, Ruixiao Li, Zhijian Zhou +7
Can Graph Neural Networks Help Logic Reasoning?
Yuyu Zhang, Xinshi Chen, Yuan Yang +4
Structure-aware Semantic Discrepancy and Consistency for 3D Medical Image Self-supervised Learning
Tan Pan, Zhaorui Tan, Kaiyu Guo +6
GeniePath: Graph Neural Networks with Adaptive Receptive Paths
Ziqi Liu, Chaochao Chen, Longfei Li +4
DinTucker: Scaling up Gaussian process models on multidimensional arrays with billions of elements
Shandian Zhe, Yuan Qi, Youngja Park +2
AGL: a Scalable System for Industrial-purpose Graph Machine Learning
Dalong Zhang, Xin Huang, Ziqi Liu +8
SpellGCN: Incorporating Phonological and Visual Similarities into Language Models for Chinese Spelling Check
Xingyi Cheng, Weidi Xu, Kunlong Chen +5
Policy Learning with $α$-Expected Welfare
Yanqin Fan, Yuan Qi, Gaoqian Xu
Practical Privacy Preserving POI Recommendation
Chaochao Chen, Jun Zhou, Bingzhe Wu +4
Double Neural Counterfactual Regret Minimization
Hui Li, Kailiang Hu, Zhibang Ge +3
Finding Kissing Numbers with Game-theoretic Reinforcement Learning
Chengdong Ma, Théo Tao Zhaowei, Pengyu Li +7
Infinity Parser: Layout Aware Reinforcement Learning for Scanned Document Parsing
Baode Wang, Biao Wu, Weizhen Li +8
Generating Natural Language Adversarial Examples on a Large Scale with Generative Models
Yankun Ren, Jianbin Lin, Siliang Tang +4
PA-Net: Precipitation-Adaptive Mixture-of-Experts for Long-Tail Rainfall Nowcasting
Xinyu Xiao, Sen Lei, Eryun Liu +5
Monkey King Bang: A Unified Scientific Multimodal Foundation Model
Hesen Chen, Xinyu Su, Xiaomeng Yang +11
AURORA:Automated Training Framework of Universal Process Reward Models via Ensemble Prompting and Reverse Verification
Xiaoyu Tan, Tianchu Yao, Chao Qu +8
AlphaFolding: 4D Diffusion for Dynamic Protein Structure Prediction with Reference and Motion Guidance
Kaihui Cheng, Ce Liu, Qingkun Su +6
Universality Reconsidered: Rethinking the Validation of Foundation Models for General-Purpose 3D Medical Segmentation
Yichi Zhang, Feiyang Xiao, Le Xue +6
Equivariant Asynchronous Diffusion: An Adaptive Denoising Schedule for Accelerated Molecular Conformation Generation
Junyi An, Chao Qu, Yun-Fei Shi +3
SegAnyPET: Universal Promptable Segmentation from Positron Emission Tomography Images
Yichi Zhang, Le Xue, Wenbo Zhang +5
Variational Policy Propagation for Multi-agent Reinforcement Learning
Chao Qu, Hui Li, Chang Liu +6
EigenGP: Sparse Gaussian process models with data-dependent eigenfunctions
Yuan Qi, Bo Dai, Yao Zhu
Imagen 3
Imagen-Team-Google, :, Jason Baldridge +257
CTkvr: KV Cache Retrieval for Long-Context LLMs via Centroid then Token Indexing
Kuan Lu, Shuhang Lin, Sai Wu +7
DyJR: Preserving Diversity in Reinforcement Learning with Verifiable Rewards via Dynamic Jensen-Shannon Replay
Long Li, Zhijian Zhou, Tianyi Wang +7
Generative Adversarial User Model for Reinforcement Learning Based Recommendation System
Xinshi Chen, Shuang Li, Hui Li +3