Publications (121)
Learning Pore-scale Multiphase Flow from 4D Velocimetry
Chunyang Wang, Linqi Zhu, Yuxuan Gu +9
Fea2Fea: Exploring Structural Feature Correlations via Graph Neural Networks
Jiaqing Xie, Rex Ying
Rethinking Reward Supervision: Rubric-Conditioned Self-Distillation
Siyi Gu, Jialin Chen, Sophia Zhou +2
GInX-Eval: Towards In-Distribution Evaluation of Graph Neural Network Explanations
Kenza Amara, Mennatallah El-Assady, Rex Ying
HELM: Hyperbolic Large Language Models via Mixture-of-Curvature Experts
Neil He, Rishabh Anand, Hiren Madhu +5
Efficient Long-Horizon Vision-Language-Action Models via Static-Dynamic Disentanglement
Weikang Qiu, Huashuo Lei, Tinglin Huang +1
FlexRec: Adapting LLM-based Recommenders for Flexible Needs via Reinforcement Learning
Yijun Pan, Weikang Qiu, Qiyao Ma +4
Benchmarking AI Agents for Addressing Scientific Challenges Across Scales
Tianyu Liu, Allen Xin Wang, Antonia Panescu +30
Hyperbolic Deep Learning for Foundation Models: A Survey
Neil He, Hiren Madhu, Ngoc Bui +2
Representation Learning for Frequent Subgraph Mining
Rex Ying, Tianyu Fu, Andrew Wang +3
HyperCore: The Core Framework for Building Hyperbolic Foundation Models with Comprehensive Modules
Neil He, Menglin Yang, Rex Ying
TrustLLM: Trustworthiness in Large Language Models
Yue Huang, Lichao Sun, Haoran Wang +67
Hyperbolic Fine-Tuning for Large Language Models
Menglin Yang, Ram Samarth B B, Aosong Feng +4
BLIS-Net: Classifying and Analyzing Signals on Graphs
Charles Xu, Laney Goldman, Valentina Guo +7
Protein-Nucleic Acid Complex Modeling with Frame Averaging Transformer
Tinglin Huang, Zhenqiao Song, Rex Ying +1
Make Each Token Count: Towards Improving Long-Context Performance with KV Cache Eviction
Ngoc Bui, Hieu Trung Nguyen, Arman Cohan +1
Multi-Modal Time Series Prediction via Mixture of Modulated Experts
Lige Zhang, Ali Maatouk, Jialin Chen +2
Scalable Generation of Spatial Transcriptomics from Histology Images via Whole-Slide Flow Matching
Tinglin Huang, Tianyu Liu, Mehrtash Babadi +2
Bridging the Gap of AutoGraph between Academia and Industry: Analysing AutoGraph Challenge at KDD Cup 2020
Zhen Xu, Lanning Wei, Huan Zhao +4
Variational Learning for Insertion-based Generation
Yangtian Zhang, Zhe Wang, Arthur Gretton +4
Flow Matching for Collaborative Filtering
Chengkai Liu, Yangtian Zhang, Jianling Wang +2
Omanic: Towards Step-wise Evaluation of Multi-hop Reasoning in Large Language Models
Xiaojie Gu, Sherry T. Tong, Aosong Feng +8
RNA-FrameFlow: Flow Matching for de novo 3D RNA Backbone Design
Rishabh Anand, Chaitanya K. Joshi, Alex Morehead +7
The Heterophilic Graph Learning Handbook: Benchmarks, Models, Theoretical Analysis, Applications and Challenges
Sitao Luan, Chenqing Hua, Qincheng Lu +11
TempME: Towards the Explainability of Temporal Graph Neural Networks via Motif Discovery
Jialin Chen, Rex Ying
MuSe-GNN: Learning Unified Gene Representation From Multimodal Biological Graph Data
Tianyu Liu, Yuge Wang, Rex Ying +1
HiPool: Modeling Long Documents Using Graph Neural Networks
Irene Li, Aosong Feng, Dragomir Radev +1
Explaining Graph Neural Networks via Structure-aware Interaction Index
Ngoc Bui, Hieu Trung Nguyen, Viet Anh Nguyen +1
Tele-LLMs: A Series of Specialized Large Language Models for Telecommunications
Ali Maatouk, Kenny Chirino Ampudia, Rex Ying +1
GraphFramEx: Towards Systematic Evaluation of Explainability Methods for Graph Neural Networks
Kenza Amara, Rex Ying, Zitao Zhang +5
Efficient Automatic Machine Learning via Design Graphs
Shirley Wu, Jiaxuan You, Jure Leskovec +1
Position-aware Graph Neural Networks
Jiaxuan You, Rex Ying, Jure Leskovec
Towards Understanding Sensitive and Decisive Patterns in Explainable AI: A Case Study of Model Interpretation in Geometric Deep Learning
Jiajun Zhu, Siqi Miao, Rex Ying +1
Online Detection of Anomalies in Temporal Knowledge Graphs with Interpretability
Jiasheng Zhang, Rex Ying, Jie Shao
Cache What Lasts: Token Retention for Memory-Bounded KV Cache in LLMs
Ngoc Bui, Shubham Sharma, Simran Lamba +2
Position: Beyond Euclidean -- Foundation Models Should Embrace Non-Euclidean Geometries
Neil He, Jiahong Liu, Buze Zhang +6
Mixture-of-Personas Language Models for Population Simulation
Ngoc Bui, Hieu Trung Nguyen, Shantanu Kumar +4
HiPoNet: A Multi-View Simplicial Complex Network for High Dimensional Point-Cloud and Single-Cell Data
Siddharth Viswanath, Hiren Madhu, Dhananjay Bhaskar +7
Efficient High-Resolution Time Series Classification via Attention Kronecker Decomposition
Aosong Feng, Jialin Chen, Juan Garza +5
Learning to Simulate Complex Physics with Graph Networks
Alvaro Sanchez-Gonzalez, Jonathan Godwin, Tobias Pfaff +3
Breaking Information Cocoons: A Hyperbolic Framework for Balancing Exploration and Exploitation in Recommender Systems
Qiyao Ma, Menglin Yang, Mingxuan Ju +3
Non-Markovian Discrete Diffusion with Causal Language Models
Yangtian Zhang, Sizhuang He, Daniel Levine +7
Graph Ensemble Learning over Multiple Dependency Trees for Aspect-level Sentiment Classification
Xiaochen Hou, Peng Qi, Guangtao Wang +4
LitBench: A Graph-Centric Large Language Model Benchmarking Tool For Literature Tasks
Andreas Varvarigos, Ali Maatouk, Jiasheng Zhang +4
GNNExplainer: Generating Explanations for Graph Neural Networks
Rex Ying, Dylan Bourgeois, Jiaxuan You +2
Fin-RATE: A Real-world Financial Analytics and Tracking Evaluation Benchmark for LLMs on SEC Filings
Yidong Jiang, Junrong Chen, Eftychia Makri +7
Hypformer: Exploring Efficient Transformer Fully in Hyperbolic Space
Menglin Yang, Harshit Verma, Delvin Ce Zhang +3
Adversarially Robust Neural Architecture Search for Graph Neural Networks
Beini Xie, Heng Chang, Ziwei Zhang +5
Steerable Instruction Following Coding Data Synthesis with Actor-Parametric Schema Co-Evolution
Tinglin Huang, Bo Chen, Xiao Zhang +2
CoPersona: Collaborative Persona Graphs for Robust LLM Personalization
Yangtian Zhang, Leyao Wang, Hiren Madhu +3
Design Space for Graph Neural Networks
Jiaxuan You, Rex Ying, Jure Leskovec
Reaction-conditioned De Novo Enzyme Design with GENzyme
Chenqing Hua, Jiarui Lu, Yong Liu +7
From Similarity to Superiority: Channel Clustering for Time Series Forecasting
Jialin Chen, Jan Eric Lenssen, Aosong Feng +5
Seeing Through the Brain: New Insights from Decoding Visual Stimuli with fMRI
Zheng Huang, Enpei Zhang, Weikang Qiu +7
GRIL: Knowledge Graph Retrieval-Integrated Learning with Large Language Models
Jialin Chen, Houyu Zhang, Seongjun Yun +6
TRACE: Grounding Time Series in Context for Multimodal Embedding and Retrieval
Jialin Chen, Ziyu Zhao, Gaukhar Nurbek +5
MUDiff: Unified Diffusion for Complete Molecule Generation
Chenqing Hua, Sitao Luan, Minkai Xu +4
Towards A Universal Graph Structural Encoder
Jialin Chen, Haolan Zuo, Haoyu Peter Wang +3
Reasoning through Verifiable Forecast Actions: Consistency-Grounded RL for Financial LLMs
Jialin Chen, Aosong Feng, Harshit Verma +7
Neural Distance Embeddings for Biological Sequences
Gabriele Corso, Rex Ying, Michal Pándy +3
FOLIO: Natural Language Reasoning with First-Order Logic
Simeng Han, Hailey Schoelkopf, Yilun Zhao +32
FusionRetro: Molecule Representation Fusion via In-Context Learning for Retrosynthetic Planning
Songtao Liu, Zhengkai Tu, Minkai Xu +6
Low-Rank Adaptation for Foundation Models: A Comprehensive Review
Menglin Yang, Jialin Chen, Jinkai Tao +9
DTGB: A Comprehensive Benchmark for Dynamic Text-Attributed Graphs
Jiasheng Zhang, Jialin Chen, Menglin Yang +4
Learning to Group Auxiliary Datasets for Molecule
Tinglin Huang, Ziniu Hu, Rex Ying
Neural Execution of Graph Algorithms
Petar VeliÄkoviÄ, Rex Ying, Matilde Padovano +2
SRTJ: Self-Evolving Rule-Driven Training-Free LLM Jailbreaking
Jindong Li, Ying Liu, Yali Fu +4
Personalized Safety Alignment for Text-to-Image Diffusion Models
Yu Lei, Jinbin Bai, Qingyu Shi +4
MindLLM: A Subject-Agnostic and Versatile Model for fMRI-to-Text Decoding
Weikang Qiu, Zheng Huang, Haoyu Hu +3
Redundancy-Free Computation Graphs for Graph Neural Networks
Zhihao Jia, Sina Lin, Rex Ying +3
Learning High-Order Relationships of Brain Regions
Weikang Qiu, Huangrui Chu, Selena Wang +4
Improving Graph Attention Networks with Large Margin-based Constraints
Guangtao Wang, Rex Ying, Jing Huang +1
Graph Convolutional Neural Networks for Web-Scale Recommender Systems
Rex Ying, Ruining He, Kaifeng Chen +3
Representation Learning on Graphs: Methods and Applications
William L. Hamilton, Rex Ying, Jure Leskovec
GraphRNN: Generating Realistic Graphs with Deep Auto-regressive Models
Jiaxuan You, Rex Ying, Xiang Ren +2
Hyperbolic Graph Convolutional Neural Networks
Ines Chami, Rex Ying, Christopher Ré +1
Thought Propagation: An Analogical Approach to Complex Reasoning with Large Language Models
Junchi Yu, Ran He, Rex Ying
P-FOLIO: Evaluating and Improving Logical Reasoning with Abundant Human-Written Reasoning Chains
Simeng Han, Aaron Yu, Rui Shen +13
Long Sequence Modeling with Attention Tensorization: From Sequence to Tensor Learning
Aosong Feng, Rex Ying, Leandros Tassiulas
Platonic Transformers: A Solid Choice For Equivariance
Mohammad Mohaiminul Islam, Rishabh Anand, David R. Wessels +7
Multi-hop Attention Graph Neural Network
Guangtao Wang, Rex Ying, Jing Huang +1
Experience Retrieval-Augmentation with Electronic Health Records Enables Accurate Discharge QA
Justice Ou, Tinglin Huang, Yilun Zhao +3
Towards Non-Euclidean Foundation Models: Advancing AI Beyond Euclidean Frameworks
Menglin Yang, Yifei Zhang, Jialin Chen +2
Parameter-Efficient Fine-Tuning of LLMs with Mixture of Space Experts
Buze Zhang, Jinkai Tao, Zilang Zeng +4
Beyond Feature Importance: Feature Interactions in Predicting Post-Stroke Rigidity with Graph Explainable AI
Jiawei Xu, Yonggeon Lee, Anthony Elkommos Youssef +6
Zatom-1: Towards a Multimodal Foundation Model for 3D Molecules and Materials
Alex Morehead, Miruna Cretu, Antonia Panescu +14
An Item is Worth a Prompt: Versatile Image Editing with Disentangled Control
Aosong Feng, Weikang Qiu, Jinbin Bai +5
HypRAG: Hyperbolic Dense Retrieval for Retrieval Augmented Generation
Hiren Madhu, Ngoc Bui, Ali Maatouk +6
Approximating Dynamic Time Warping and Edit Distance for a Pair of Point Sequences
Pankaj K. Agarwal, Kyle Fox, Jiangwei Pan +1
LitFM: A Retrieval Augmented Structure-aware Foundation Model For Citation Graphs
Jiasheng Zhang, Jialin Chen, Ali Maatouk +6
Time to REFLECT: Can We Trust LLM Judges for Evidence-based Research Agents?
Leyao Wang, Yanan He, Peng Chen +5
Learning Along the Arrow of Time: Hyperbolic Geometry for Backward-Compatible Representation Learning
Ngoc Bui, Menglin Yang, Runjin Chen +5
How Powerful is Implicit Denoising in Graph Neural Networks
Songtao Liu, Rex Ying, Hanze Dong +3
SANDWICH: Towards an Offline, Differentiable, Fully-Trainable Wireless Neural Ray-Tracing Surrogate
Yifei Jin, Ali Maatouk, Sarunas Girdzijauskas +3
BatchSampler: Sampling Mini-Batches for Contrastive Learning in Vision, Language, and Graphs
Zhen Yang, Tinglin Huang, Ming Ding +5
Implicit Reasoning in Large Language Models: A Comprehensive Survey
Jindong Li, Yali Fu, Li Fan +6
Hyperbolic Representation Learning: Revisiting and Advancing
Menglin Yang, Min Zhou, Rex Ying +2
Learning Large-scale Subsurface Simulations with a Hybrid Graph Network Simulator
Tailin Wu, Qinchen Wang, Yinan Zhang +7
Lorentzian Residual Neural Networks
Neil He, Menglin Yang, Rex Ying
Understanding and Mitigating Hyperbolic Dimensional Collapse in Graph Contrastive Learning
Yifei Zhang, Hao Zhu, Menglin Yang +4