Accelerating the 3D Elastic Wave Forward Modeling on GPU and MIC. - Yang You 001; Haohuan Fu; Xiaomeng Huang; Guojie Song; Lin Gan; Wenjian Yu; Guangwen Yang (IPDPS Workshops, 2013)

Scaling and analyzing the stencil performance on multi-core and many-core architectures. - Lin Gan; Haohuan Fu; Wei Xue 003; Yangtong Xu; Chao Yang 002; Xinliang Wang; Zihong Lv; Yang You 001; Guangwen Yang; Kaijian Ou (ICPADS, 2014)

Designing a Heuristic Cross-Architecture Combination for Breadth-First Search. - Yang You 001; David A. Bader; Maryam Mehri Dehnavi (ICPP, 2014)

An adaptive cross-architecture combination method for graph traversal. - Yang You 001; Shuaiwen Leon Song; Darren J. Kerbyson (ICS, 2014)

MIC-SVM: Designing a Highly Efficient Support Vector Machine for Advanced Modern Multi-core and Many-Core Architectures. - Yang You 001; Shuaiwen Leon Song; Haohuan Fu; Andres Marquez 001; Maryam Mehri Dehnavi; Kevin J. Barker; Kirk W. Cameron; Amanda Peters Randles; Guangwen Yang (IPDPS, 2014)

Evaluating multi-core and many-core architectures through accelerating the three-dimensional Lax-Wendroff correction stencil. - Yang You 001; Haohuan Fu; Shuaiwen Leon Song; Maryam Mehri Dehnavi; Lin Gan; Xiaomeng Huang; Guangwen Yang (Int. J. High Perform. Comput. Appl., 2014)

CA-SVM: Communication-Avoiding Support Vector Machines on Distributed Systems. - Yang You 001; James Demmel; Kenneth Czechowski; Le Song; Richard W. Vuduc (IPDPS, 2015)

Scaling Support Vector Machines on modern HPC platforms. - Yang You 001; Haohuan Fu; Shuaiwen Leon Song; Amanda Peters Randles; Darren J. Kerbyson; Andres Marquez 001; Guangwen Yang; Adolfy Hoisie (J. Parallel Distributed Comput., 2015)

Asynchronous Parallel Greedy Coordinate Descent. - Yang You 001; Xiangru Lian; Ji Liu 002; Hsiang-Fu Yu; Inderjit S. Dhillon; James Demmel; Cho-Jui Hsieh (NIPS, 2016)

Runtime Data Layout Scheduling for Machine Learning Dataset. - Yang You 001; James Demmel (ICPP, 2017)

Scaling deep learning on GPU and knights landing clusters. - Yang You 001; Aydin Buluç; James Demmel (SC, 2017)

100-epoch ImageNet Training with AlexNet in 24 Minutes. - Yang You 001; Zhao Zhang 007; Cho-Jui Hsieh; James Demmel (CoRR, 2017)

Designing and implementing a heuristic cross-architecture combination for graph traversal. - Yang You 001; Haohuan Fu; David A. Bader; Guangwen Yang (J. Parallel Distributed Comput., 2017)

Parallel Multiclass Support Vector Machine for Remote Sensing Data Classification on Multicore and Many-Core Architectures. - Weijia Li; Haohuan Fu; Yang You 001; Le Yu 001; Jiarui Fang (IEEE J. Sel. Top. Appl. Earth Obs. Remote. Sens., 2017)

Design and Implementation of a Communication-Optimal Classifier for Distributed Kernel Support Vector Machines. - Yang You 001; James Demmel; Kent Czechowski; Le Song; Rich Vuduc (IEEE Trans. Parallel Distributed Syst., 2017)

ImageNet Training in Minutes. - Yang You 001; Zhao Zhang 007; Cho-Jui Hsieh; James Demmel; Kurt Keutzer (ICPP, 2018)

Accurate, Fast and Scalable Kernel Ridge Regression on Parallel and Distributed Systems. - Yang You 001; James Demmel; Cho-Jui Hsieh; Richard W. Vuduc (ICS, 2018)

Fast LSTM Inference by Dynamic Decomposition on Cloud Systems. - Yang You 001; Yuxiong He; Samyam Rajbhandari; Wenhan Wang; Cho-Jui Hsieh; Kurt Keutzer; James Demmel (ICDM, 2019)

Large-batch training for LSTM and beyond. - Yang You 001; Jonathan Hseu; Chris Ying; James Demmel; Kurt Keutzer; Cho-Jui Hsieh (SC, 2019)

Reducing BERT Pre-Training Time from 3 Days to 76 Minutes. - Yang You 001; Jing Li; Jonathan Hseu; Xiaodan Song; James Demmel; Cho-Jui Hsieh (CoRR, 2019)

Fast Deep Neural Network Training on Distributed Systems and Cloud TPUs. - Yang You 001; Zhao Zhang 007; Cho-Jui Hsieh; James Demmel; Kurt Keutzer (IEEE Trans. Parallel Distributed Syst., 2019)

Rethinking the Value of Asynchronous Solvers for Distributed Deep Learning. - Arissa Wongpanich; Yang You 001; James Demmel (HPC Asia, 2020)

Large Batch Optimization for Deep Learning: Training BERT in 76 minutes. - Yang You 001; Jing Li; Sashank J. Reddi; Jonathan Hseu; Sanjiv Kumar; Srinadh Bhojanapalli; Xiaodan Song; James Demmel; Kurt Keutzer; Cho-Jui Hsieh (ICLR, 2020)

The Limit of the Batch Size. - Yang You 001; Yuhui Wang; Huan Zhang 001; Zhao Zhang 007; James Demmel; Cho-Jui Hsieh (CoRR, 2020)

How much progress have we made in neural network training? A New Evaluation Protocol for Benchmarking Optimizers. - Yuanhao Xiong; Xuanqing Liu; Li-Cheng Lan; Yang You 001; Si Si; Cho-Jui Hsieh (CoRR, 2020)

Fast LSTM by dynamic decomposition on cloud and distributed systems. - Yang You 001; Yuxiong He; Samyam Rajbhandari; Wenhan Wang; Cho-Jui Hsieh; Kurt Keutzer; James Demmel (Knowl. Inf. Syst., 2020)

Fast and Accurate Machine Learning on Distributed Systems and Supercomputers - Yang You 001 (2020)

Mask Aware Network for Masked Face Recognition in the Wild. - Kai Wang 036; Shuo Wang 001; Jianfei Yang; Xiaobo Wang; Baigui Sun; Hao Li 030; Yang You 001 (ICCVW, 2021)

Training EfficientNets at Supercomputer Scale: 83% ImageNet Top-1 Accuracy in One Hour. - Arissa Wongpanich; Hieu Pham 001; James Demmel; Mingxing Tan; Quoc V. Le; Yang You 001; Sameer Kumar (IPDPS Workshops, 2021)

Dynamic scaling for low-precision learning. - Ruobing Han; Min Si; James Demmel; Yang You 001 (PPoPP, 2021)

Online evolutionary batch size orchestration for scheduling deep learning workloads in GPU clusters. - Zhengda Bian; Shenggui Li; Wei Wang; Yang You 001 (SC, 2021)

Auto-Precision Scaling for Distributed Deep Learning. - Ruobing Han; James Demmel; Yang You 001 (ISC, 2021)

Communication-avoiding kernel ridge regression on parallel and distributed systems. - Yang You 001; Jingyue Huang; Cho-Jui Hsieh; Richard W. Vuduc; James Demmel (CCF Trans. High Perform. Comput., 2021)

An Efficient 2D Method for Training Super-Large Deep Learning Models. - Qifan Xu; Shenggui Li; Chaoyu Gong; Yang You 001 (CoRR, 2021)

An Efficient Training Approach for Very Large Scale Face Recognition. - Kai Wang 036; Shuo Wang 001; Zhipeng Zhou; Xiaobo Wang; Xiaojiang Peng; Baigui Sun; Hao Li 030; Yang You 001 (CoRR, 2021)

Sequence Parallelism: Making 4D Parallelism Possible. - Shenggui Li; Fuzhao Xue; Yongbin Li; Yang You 001 (CoRR, 2021)

Maximizing Parallelism in Distributed Training for Huge Neural Networks. - Zhengda Bian; Qifan Xu; Boxiang Wang; Yang You 001 (CoRR, 2021)

2.5-dimensional distributed model training. - Boxiang Wang; Qifan Xu; Zhengda Bian; Yang You 001 (CoRR, 2021)

PatrickStar: Parallel Training of Pre-trained Models via a Chunk-based Memory Management. - Jiarui Fang; Yang Yu; Shenggui Li; Yang You 001; Jie Zhou 016 (CoRR, 2021)

Sparse-MLP: A Fully-MLP Architecture with Conditional Computation. - Yuxuan Lou; Fuzhao Xue; Zangwei Zheng; Yang You 001 (CoRR, 2021)

Colossal-AI: A Unified Deep Learning System For Large-Scale Parallel Training. - Zhengda Bian; Hongxin Liu; Boxiang Wang; Haichen Huang; Yongbin Li; Chuanrui Wang; Fan Cui; Yang You 001 (CoRR, 2021)

Large-Scale Deep Learning Optimizations: A Comprehensive Survey. - Xiaoxin He; Fuzhao Xue; Xiaozhe Ren; Yang You 001 (CoRR, 2021)

Evidential instance selection for K-nearest neighbor classification of big data. - Chaoyu Gong; Zhi-gang Su; Pei-hong Wang; Qian Wang; Yang You 001 (Int. J. Approx. Reason., 2021)

Go Wider Instead of Deeper. - Fuzhao Xue; Ziji Shi; Futao Wei; Yuxuan Lou; Yong Liu 020; Yang You 001 (AAAI, 2022)

Distributed EK-NN Classification. - Chaoyu Gong; Zhi-gang Su; Qian Wang; Yang You 001 (BELIEF, 2022)

Towards Efficient and Scalable Sharpness-Aware Minimization. - Yong Liu 020; Siqi Mai; Xiangning Chen; Cho-Jui Hsieh; Yang You 001 (CVPR, 2022)

Crafting Better Contrastive Views for Siamese Representation Learning. - Xiangyu Peng; Kai Wang 036; Zheng Zhu; Mang Wang; Yang You 001 (CVPR, 2022)

An Efficient Training Approach for Very Large Scale Face Recognition. - Kai Wang 036; Shuo Wang 001; Panpan Zhang; Zhipeng Zhou; Zheng Zhu; Xiaobo Wang; Xiaojiang Peng; Baigui Sun; Hao Li 030; Yang You 001 (CVPR, 2022)

CAFE: Learning to Condense Dataset by Aligning Features. - Kai Wang 036; Bo Zhao; Xiangyu Peng; Zheng Zhu; Shuo Yang 006; Shuo Wang 001; Guan Huang 003; Hakan Bilen; Xinchao Wang; Yang You 001 (CVPR, 2022)

Modeling Motion with Multi-Modal Features for Text-Based Video Segmentation. - Wangbo Zhao; Kai Wang 036; Xiangxiang Chu; Fuzhao Xue; Xinchao Wang; Yang You 001 (CVPR, 2022)

Self-reconstructive evidential clustering for high-dimensional data. - Chaoyu Gong; Yongbin Liu; Di Fu; Yong Liu 020; Pei-hong Wang; Yang You 001 (ICDE, 2022)

Joint Evidential $K$-Nearest Neighbor Classification. - Chaoyu Gong; Yongbin Li; Yong Liu 020; Pei-hong Wang; Yang You 001 (ICDE, 2022)

Concurrent Adversarial Learning for Large-Batch Training. - Yong Liu 020; Xiangning Chen; Minhao Cheng; Cho-Jui Hsieh; Yang You 001 (ICLR, 2022)

Tesseract: Parallelize the Tensor Parallelism Efficiently. - Boxiang Wang; Qifan Xu; Zhengda Bian; Yang You 001 (ICPP, 2022)

Handling heavy-tailed input of transformer inference on GPUs. - Jiangsu Du; Jiazhi Jiang; Yang You 001; Dan Huang 001; Yutong Lu (ICS, 2022)

Random Sharpness-Aware Minimization. - Yong Liu 020; Siqi Mai; Minhao Cheng; Xiangning Chen; Cho-Jui Hsieh; Yang You 001 (NeurIPS, 2022)

Crafting Better Contrastive Views for Siamese Representation Learning. - Xiangyu Peng; Kai Wang 036; Zheng Zhu; Yang You 001 (CoRR, 2022)

Sky Computing: Accelerating Geo-distributed Computing in Federated Learning. - Jie Zhu; Shenggui Li; Yang You 001 (CoRR, 2022)

FastFold: Reducing AlphaFold Training Time from 11 Days to 67 Hours. - Shenggan Cheng; Ruidong Wu; Zhongming Yu; Bin-Rui Li; Xiwen Zhang; Jian Peng 001; Yang You 001 (CoRR, 2022)

CowClip: Reducing CTR Prediction Model Training Time from 12 hours to 10 minutes on 1 GPU. - Zangwei Zheng; Pengtai Xu; Xuan Zou; Da Tang; Zhen Li 026; Chenguang Xi; Peng Wu; Leqi Zou; Yijie Zhu; Ming Chen; Xiangzhuo Ding; Fuzhao Xue; et al. (CoRR, 2022)

Reliable Label Correction is a Good Booster When Learning with Extremely Noisy Labels. - Kai Wang 036; Xiangyu Peng; Shuo Yang 006; Jianfei Yang; Zheng Zhu; Xinchao Wang; Yang You 001 (CoRR, 2022)

Deeper vs Wider: A Revisit of Transformer Configuration. - Fuzhao Xue; Jianghai Chen; Aixin Sun; Xiaozhe Ren; Zangwei Zheng; Xiaoxin He; Xin Jiang 002; Yang You 001 (CoRR, 2022)

FaceMAE: Privacy-Preserving Face Recognition via Masked Autoencoders. - Kai Wang 036; Bo Zhao; Xiangyu Peng; Zheng Zhu; Jiankang Deng; Xinchao Wang; Hakan Bilen; Yang You 001 (CoRR, 2022)

A Frequency-aware Software Cache for Large Recommendation System Embeddings. - Jiarui Fang; Geng Zhang; Jiatong Han; Shenggui Li; Zhengda Bian; Yongbin Li; Jin Liu; Yang You 001 (CoRR, 2022)

Prompt Vision Transformer for Domain Generalization. - Zangwei Zheng; Xiangyu Yue 001; Kai Wang 036; Yang You 001 (CoRR, 2022)

EnergonAI: An Inference System for 10-100 Billion Parameter Transformer Models. - Jiangsu Du; Ziming Liu; Jiarui Fang; Shenggui Li; Yongbin Li; Yutong Lu; Yang You 001 (CoRR, 2022)

Elixir: Train a Large Language Model on a Small GPU Cluster. - Haichen Huang; Jiarui Fang; Hongxin Liu; Shenggui Li; Yang You 001 (CoRR, 2022)

Distributed evidential clustering toward time series with big data issue. - Chaoyu Gong; Zhi-gang Su; Pei-hong Wang; Yang You 001 (Expert Syst. Appl., 2022)

Weakly Supervised Learning for Textbook Question Answering. - Jie Ma 001; Qi Chai; Jingyue Huang; Jun Liu 002; Yang You 001; Qinghua Zheng (IEEE Trans. Image Process., 2022)

GPTR: Gestalt-Perception Transformer for Diagram Object Detection. - Xin Hu; Lingling Zhang 005; Jun Liu 002; Jinfu Fan; Yang You 001; Yaqiang Wu (AAAI, 2023)

CowClip: Reducing CTR Prediction Model Training Time from 12 Hours to 10 Minutes on 1 GPU. - Zangwei Zheng; Pengtai Xu; Xuan Zou; Da Tang; Zhen Li 026; Chenguang Xi; Peng Wu; Leqi Zou; Yijie Zhu; Ming Chen; Xiangzhuo Ding; Fuzhao Xue; et al. (AAAI, 2023)

Sequence Parallelism: Long Sequence Training from System Perspective. - Shenggui Li; Fuzhao Xue; Chaitanya Baranwal; Yongbin Li; Yang You 001 (ACL (1), 2023)

CAME: Confidence-guided Adaptive Memory Efficient Optimization. - Yang Luo; Xiaozhe Ren; Zangwei Zheng; Zhuo Jiang; Xin Jiang 002; Yang You 001 (ACL (1), 2023)

MSINet: Twins Contrastive Search of Multi-Scale Interaction for Object ReID. - Jianyang Gu; Kai Wang 036; Hao Luo 004; Chen Chen 114; Wei Jiang 009; Yuqiang Fang; Shanghang Zhang; Yang You 001; Jian Zhao 006 (CVPR, 2023)

BiCro: Noisy Correspondence Rectification for Multi-modality Data via Bi-directional Cross-modal Similarity Consistency. - Shuo Yang 006; Zhaopan Xu; Kai Wang 036; Yang You 001; Hongxun Yao; Tongliang Liu; Min Xu 001 (CVPR, 2023)

DREAM: Efficient Dataset Distillation by Representative Matching. - Yanqing Liu; Jianyang Gu; Kai Wang 036; Zheng Zhu; Wei Jiang 009; Yang You 001 (ICCV, 2023)

Preventing Zero-Shot Transfer Degradation in Continual Learning of Vision-Language Models. - Zangwei Zheng; Mingyuan Ma; Kai Wang 036; Ziheng Qin; Xiangyu Yue 001; Yang You 001 (ICCV, 2023)

Dataset Quantization. - Daquan Zhou; Kai Wang 036; Jianyang Gu; Xiangyu Peng; Dongze Lian; Yifan Zhang 004; Yang You 001; Jiashi Feng (ICCV, 2023)

Hierarchical Dialogue Understanding with Special Tokens and Turn-level Attention. - Xiao Liu 035; Jian Zhang; Heng Zhang; Fuzhao Xue; Yang You 001 (Tiny Papers @ ICLR, 2023)

One Student Knows All Experts Know: From Sparse to Dense. - Fuzhao Xue; Xiaoxin He; Xiaozhe Ren; Yuxuan Lou; Yang You 001 (Tiny Papers @ ICLR, 2023)

Divide to Adapt: Mitigating Confirmation Bias for Domain Adaptation of Black-Box Predictors. - Jianfei Yang; Xiangyu Peng; Kai Wang 036; Zheng Zhu; Jiashi Feng; Lihua Xie 001; Yang You 001 (ICLR, 2023)

A Study on Transformer Configuration and Training Objective. - Fuzhao Xue; Jianghai Chen; Aixin Sun; Xiaozhe Ren; Zangwei Zheng; Xiaoxin He; Yongming Chen; Xin Jiang 002; Yang You 001 (ICML, 2023)

Adaptive Computation with Elastic Input Sequence. - Fuzhao Xue; Valerii Likhosherstov; Anurag Arnab; Neil Houlsby; Mostafa Dehghani 001; Yang You 001 (ICML, 2023)

Colossal-AI: A Unified Deep Learning System For Large-Scale Parallel Training. - Shenggui Li; Hongxin Liu; Zhengda Bian; Jiarui Fang; Haichen Huang; Yuliang Liu; Boxiang Wang; Yang You 001 (ICPP, 2023)

An Efficient 2D Method for Training Super-Large Deep Learning Models. - Qifan Xu; Yang You 001 (IPDPS, 2023)

To Repeat or Not To Repeat: Insights from Scaling LLM under Token-Crisis. - Fuzhao Xue; Yao Fu; Wangchunshu Zhou; Zangwei Zheng; Yang You 001 (NeurIPS, 2023)

Does Graph Distillation See Like Vision Dataset Counterpart? - Beining Yang; Kai Wang 036; Qingyun Sun; Cheng Ji 001; Xingcheng Fu; Hao Tang 005; Yang You 001; Jianxin Li 002 (NeurIPS, 2023)

Response Length Perception and Sequence Scheduling: An LLM-Empowered LLM Inference Pipeline. - Zangwei Zheng; Xiaozhe Ren; Fuzhao Xue; Yang Luo; Xin Jiang 002; Yang You 001 (NeurIPS, 2023)

Hanayo: Harnessing Wave-like Pipeline Parallelism for Enhanced Large Model Training Efficiency. - Ziming Liu; Shenggan Cheng; Haotian Zhou; Yang You 001 (SC, 2023)

ATP: Adaptive Tensor Parallelism for Foundation Models. - Shenggan Cheng; Ziming Liu; Jiangsu Du; Yang You 001 (CoRR, 2023)

Colossal-Auto: Unified Automation of Parallelization and Activation Checkpoint for Large-scale Models. - Yuliang Liu; Shenggui Li; Jiarui Fang; Yanjun Shao; Boyuan Yao; Yang You 001 (CoRR, 2023)

DiM: Distilling Dataset into Generative Model. - Kai Wang 036; Jianyang Gu; Daquan Zhou; Zheng Zhu; Wei Jiang 009; Yang You 001 (CoRR, 2023)

InfoBatch: Lossless Training Speed Up by Unbiased Dynamic Data Pruning. - Ziheng Qin; Kai Wang 036; Zangwei Zheng; Jianyang Gu; Xiangyu Peng; Daquan Zhou; Yang You 001 (CoRR, 2023)

Summarizing Stream Data for Memory-Restricted Online Continual Learning. - Jianyang Gu; Kai Wang 036; Wei Jiang 009; Yang You 001 (CoRR, 2023)

Learning Referring Video Object Segmentation from Weak Annotation. - Wangbo Zhao; Kepan Nan; Songyang Zhang; Kai Chen 026; Dahua Lin; Yang You 001 (CoRR, 2023)

The Snowflake Hypothesis: Training Deep GNN with One Node One Receptive field. - Kun Wang 042; Guohao Li 001; Shilong Wang 002; Guibin Zhang; Kai Wang 036; Yang You 001; Xiaojiang Peng; Yuxuan Liang; Yang Wang 015 (CoRR, 2023)

Color Prompting for Data-Free Continual Unsupervised Domain Adaptive Person Re-Identification. - Jianyang Gu; Hao Luo 004; Kai Wang 036; Wei Jiang 009; Yang You 001; Jian Zhao 006 (CoRR, 2023)

Boosting Unsupervised Contrastive Learning Using Diffusion-Based Data Augmentation From Scratch. - Zelin Zang; Hao Luo 004; Kai Wang 036; Panpan Zhang; Fan Wang 019; Stan Z. Li; Yang You 001 (CoRR, 2023)

Can pre-trained models assist in dataset distillation? - Yao Lu; Xuguang Chen; Yuchen Zhang; Jianyang Gu; Tianle Zhang; Yifan Zhang 004; Xiaoniu Yang; Qi Xuan 001; Kai Wang 036; Yang You 001 (CoRR, 2023)

Let's reward step by step: Step-Level reward model as the Navigators for Reasoning. - Qianli Ma; Haotian Zhou; Tingkai Liu; Jianbo Yuan; Pengfei Liu; Yang You 001; Hongxia Yang (CoRR, 2023)

LoBaSS: Gauging Learnability in Supervised Fine-tuning Data. - Haotian Zhou; Tingkai Liu; Qianli Ma; Jianbo Yuan; Pengfei Liu; Yang You 001; Hongxia Yang (CoRR, 2023)

DREAM+: Efficient Dataset Distillation by Bidirectional Representative Matching. - Yanqing Liu; Jianyang Gu; Kai Wang 036; Zheng Zhu; Kaipeng Zhang; Wei Jiang 009; Yang You 001 (CoRR, 2023)

MLLMs-Augmented Visual-Language Representation Learning. - Yanqing Liu; Kai Wang 036; Wenqi Shao; Ping Luo 002; Yu Qiao 001; Mike Zheng Shou; Kaipeng Zhang; Yang You 001 (CoRR, 2023)

Parallel Training of Pre-Trained Models via Chunk-Based Dynamic Memory Management. - Jiarui Fang; Zilin Zhu; Shenggui Li; Hui Su; Yang Yu; Jie Zhou 016; Yang You 001 (IEEE Trans. Parallel Distributed Syst., 2023)

Multitask Learning for Visual Question Answering. - Jie Ma 001; Jun Liu 002; Qika Lin; Bei Wu; Yaxian Wang; Yang You 001 (IEEE Trans. Neural Networks Learn. Syst., 2023)

A Sparse Reconstructive Evidential K-Nearest Neighbor Classifier for High-Dimensional Data. - Chaoyu Gong; Zhi-gang Su; Pei-hong Wang; Qian Wang; Yang You 001 (IEEE Trans. Knowl. Data Eng., 2023)

Adaptive evidential K-NN classification: Integrating neighborhood search and feature weighting. - Chaoyu Gong; Zhi-gang Su; Xinyi Zhang; Yang You 001 (Inf. Sci., 2023)

Summarizing Stream Data for Memory-Constrained Online Continual Learning. - Jianyang Gu; Kai Wang 036; Wei Jiang 009; Yang You 001 (AAAI, 2024)

ParaGAN: A Scalable Distributed Training Framework for Generative Adversarial Networks. - Ziji Shi; Jialin Li 001; Yang You 001 (SoCC, 2024)

Efficient Dataset Distillation via Minimax Diffusion. - Jianyang Gu; Saeed Vahidian; Vyacheslav Kungurtsev; Haonan Wang; Wei Jiang 009; Yang You 001; Yiran Chen 001 (CVPR, 2024)

Dataset Growth. - Ziheng Qin; Zhaopan Xu; Yukun Zhou; Zangwei Zheng; Zebang Cheng; Hao Tang 005; Lei Shang; Baigui Sun; Xiaojiang Peng; Radu Timofte; Hongxun Yao; Kai Wang 036; et al. (ECCV (9), 2024)

DiM: Distilling Dataset into Generative Model. - Kai Wang 036; Jianyang Gu; Hansong Zhang 005; Daquan Zhou; Zheng Zhu; Wei Jiang 009; Yang You 001 (ECCV Workshops (19), 2024)

How Does the Textual Information Affect the Retrieval of Multimodal In-Context Learning? - Yang Luo; Zangwei Zheng; Zirui Zhu; Yang You 001 (EMNLP, 2024)

Towards Lossless Dataset Distillation via Difficulty-Aligned Trajectory Matching. - Ziyao Guo; Kai Wang 036; George Cazenavette; Hui Li; Kaipeng Zhang; Yang You 001 (ICLR, 2024)

InfoBatch: Lossless Training Speed Up by Unbiased Dynamic Data Pruning. - Ziheng Qin; Kai Wang 036; Zangwei Zheng; Jianyang Gu; Xiangyu Peng; Zhaopan Xu; Daquan Zhou; Lei Shang; Baigui Sun; Xuansong Xie; Yang You 001 (ICLR, 2024)

AutoChunk: Automated Activation Chunk for Memory-Efficient Deep Learning Inference. - Xuanlei Zhao; Shenggan Cheng; Guangyang Lu; Haotian Zhou; Bin Jia; Yang You 001 (ICLR, 2024)

OpenMoE: An Early Effort on Open Mixture-of-Experts Language Models. - Fuzhao Xue; Zian Zheng 001; Yao Fu; Jinjie Ni; Zangwei Zheng; Wangchunshu Zhou; Yang You 001 (ICML, 2024)

DiffAug: Enhance Unsupervised Contrastive Learning with Domain-Knowledge-Free Diffusion-based Data Augmentation. - Zelin Zang; Hao Luo 004; Kai Wang 036; Panpan Zhang; Fan Wang 019; Stan Z. Li; Yang You 001 (ICML, 2024)

Navigating Complexity: Toward Lossless Graph Condensation via Expanding Window Matching. - Yuchen Zhang; Tianle Zhang; Kai Wang 036; Ziyao Guo; Yuxuan Liang; Xavier Bresson; Wei Jin; Yang You 001 (ICML, 2024)

Single Domain Generalization For Scene Classification Using Style-Oriented Data Augmentation. - Yi Zhao; Guancong Lin; Juepeng Zheng; Yang You 001; Haohuan Fu (IGARSS, 2024)

The Snowflake Hypothesis: Training and Powering GNN with One Node One Receptive Field. - Kun Wang 056; Guohao Li 001; Shilong Wang 002; Guibin Zhang; Kai Wang 036; Yang You 001; Junfeng Fang; Xiaojiang Peng; Yuxuan Liang; Yang Wang 015 (KDD, 2024)

HeteGen: Efficient Heterogeneous Parallel Inference for Large Language Models on Resource-Constrained Devices. - Xuanlei Zhao; Bin Jia; Haotian Zhou; Ziming Liu; Shenggan Cheng; Yang You 001 (MLSys, 2024)

MixEval: Deriving Wisdom of the Crowd from LLM Benchmark Mixtures. - Jinjie Ni; Fuzhao Xue; Xiang Yue; Yuntian Deng; Mahir Shah; Kabir Jain; Graham Neubig; Yang You 001 (NeurIPS, 2024)

Rethinking Human Evaluation Protocol for Text-to-Video Models: Enhancing Reliability, Reproducibility, and Practicality. - Tianle Zhang; Langtian Ma; Yuchen Yan; Yuchen Zhang; Yue Yang; Ziyao Guo; Wenqi Shao; Kai Wang 036; Yang You 001; Yu Qiao 001; Ping Luo 002; Kaipeng Zhang (NeurIPS, 2024)

Dynamic Tuning Towards Parameter and Inference Efficiency for ViT Adaptation. - Wangbo Zhao; Jiasheng Tang; Yizeng Han; Yibing Song; Kai Wang 036; Gao Huang 001; Fan Wang 019; Yang You 001 (NeurIPS, 2024)

FastFold: Optimizing AlphaFold Training and Inference on GPU Clusters. - Shenggan Cheng; Xuanlei Zhao; Guangyang Lu; Jiarui Fang; Tian Zheng; Ruidong Wu; Xiwen Zhang; Jian Peng 001; Yang You 001 (PPoPP, 2024)

Helen: Optimizing CTR Prediction Models with Frequency-wise Hessian Eigenvalue Regularization. - Zirui Zhu; Yong Liu 020; Zangwei Zheng; Huifeng Guo; Yang You 001 (WWW, 2024)

Must: Maximizing Latent Capacity of Spatial Transcriptomics Data. - Zelin Zang; Liangyu Li; Yongjie Xu; Chenrui Duan; Kai Wang 036; Yang You 001; Yi Sun; Stan Z. Li (CoRR, 2024)

AutoChunk: Automated Activation Chunk for Memory-Efficient Long Sequence Inference. - Xuanlei Zhao; Shenggan Cheng; Guangyang Lu; Jiarui Fang; Haotian Zhou; Bin Jia; Ziming Liu; Yang You 001 (CoRR, 2024)

RAP: Retrieval-Augmented Planning with Contextual Memory for Multimodal LLM Agents. - Tomoyuki Kagaya; Thong Jing Yuan; Yuxuan Lou; Jayashree Karlekar; Sugiri Pranata; Akira Kinose; Koki Oguri; Felix Wick; Yang You 001 (CoRR, 2024)

Two Trades is not Baffled: Condensing Graph via Crafting Rational Gradient Matching. - Tianle Zhang; Yuchen Zhang; Kun Wang 056; Kai Wang 036; Beining Yang; Kaipeng Zhang; Wenqi Shao; Ping Liu 004; Joey Tianyi Zhou; Yang You 001 (CoRR, 2024)

Neural Network Diffusion. - Kai Wang 036; Zhaopan Xu; Yukun Zhou; Zelin Zang; Trevor Darrell; Zhuang Liu 003; Yang You 001 (CoRR, 2024)

Sparse MeZO: Less Parameters for Better Performance in Zeroth-Order LLM Fine-Tuning. - Yong Liu 020; Zirui Zhu; Chaoyu Gong; Minhao Cheng; Cho-Jui Hsieh; Yang You 001 (CoRR, 2024)

HeteGen: Heterogeneous Parallel Inference for Large Language Models on Resource-Constrained Devices. - Xuanlei Zhao; Bin Jia; Haotian Zhou; Ziming Liu; Shenggan Cheng; Yang You 001 (CoRR, 2024)

DSP: Dynamic Sequence Parallelism for Multi-Dimensional Transformers. - Xuanlei Zhao; Shenggan Cheng; Zangwei Zheng; Zheming Yang; Ziming Liu; Yang You 001 (CoRR, 2024)

Is Sora a World Simulator? A Comprehensive Survey on General World Models and Beyond. - Zheng Zhu; Xiaofeng Wang; Wangbo Zhao; Chen Min; Nianchen Deng; Min Dou; Yuqi Wang 001; Botian Shi; Kai Wang 036; Chi Zhang; Yang You 001; Zhaoxiang Zhang 001; et al. (CoRR, 2024)

A Closer Look at Time Steps is Worthy of Triple Speed-Up for Diffusion Model Training. - Kai Wang 036; Yukun Zhou; Mingjia Shi; Zhihang Yuan; Yuzhang Shang; Xiaojiang Peng; Hanwang Zhang; Yang You 001 (CoRR, 2024)

WallFacer: Guiding Transformer Model Training Out of the Long-Context Dark Forest with N-body Problem. - Ziming Liu; Shaoyu Wang; Shenggan Cheng; Zhongkai Zhao; Xuanlei Zhao; James Demmel; Yang You 001 (CoRR, 2024)

Conditional LoRA Parameter Generation. - Xiaolong Jin; Kai Wang 036; Dongwen Tang; Wangbo Zhao; Yukun Zhou; Junshu Tang; Yang You 001 (CoRR, 2024)

More Than Positive and Negative: Communicating Fine Granularity in Medical Diagnosis. - Xiangyu Peng; Kai Wang 036; Jianfei Yang; Yingying Zhu; Yang You 001 (CoRR, 2024)

Prioritize Alignment in Dataset Distillation. - Zekai Li; Ziyao Guo; Wangbo Zhao; Tianle Zhang; Zhi-Qi Cheng; Samir Khaki; Kaipeng Zhang; Ahmad Sajedi; Konstantinos N. Plataniotis; Kai Wang 036; Yang You 001 (CoRR, 2024)

Visual Perception in Text Strings. - Qi Jia; Xiang Yue; Shanshan Huang 002; Ziheng Qin; Yizhu Liu; Bill Yuchen Lin; Yang You 001 (CoRR, 2024)

MixEval-X: Any-to-Any Evaluations from Real-World Data Mixtures. - Jinjie Ni; Yifan Song; Deepanway Ghosal; Bo Li; David Junhao Zhang; Xiang Yue; Fuzhao Xue; Zian Zheng 001; Kaichen Zhang; Mahir Shah; Kabir Jain; Yang You 001; et al. (CoRR, 2024)

EnvBridge: Bridging Diverse Environments with Cross-Environment Knowledge Transfer for Embodied AI. - Tomoyuki Kagaya; Yuxuan Lou; Thong Jing Yuan; Subramanian Lakshmi; Jayashree Karlekar; Sugiri Pranata; Natsuki Murakami; Akira Kinose; Koki Oguri; Felix Wick; Yang You 001 (CoRR, 2024)

Faster Vision Mamba is Rebuilt in Minutes via Merged Token Re-training. - Mingjia Shi; Yuhao Zhou; Ruiji Yu; Zekai Li; Zhiyuan Liang; Xuanlei Zhao; Xiaojiang Peng; Tanmay Rajpurohit; Shanmukha Ramakrishna Vedantam; Wangbo Zhao; Kai Wang 036; Yang You 001 (CoRR, 2024)

Boosting LLM via Learning from Data Iteratively and Selectively. - Qi Jia; Siyu Ren; Ziheng Qin; Fuzhao Xue; Jinjie Ni; Yang You 001 (CoRR, 2024)

Open-Sora: Democratizing Efficient Video Production for All. - Zangwei Zheng; Xiangyu Peng; Tianji Yang; Chenhui Shen; Shenggui Li; Hongxin Liu; Yukun Zhou; Tianyi Li; Yang You 001 (CoRR, 2024)

Sparse Reconstructive Evidential Clustering for Multi-View Data. - Chaoyu Gong; Yang You 001 (IEEE CAA J. Autom. Sinica, 2024)

Self-filling evidential clustering for partial multi-view data. - Chaoyu Gong; Yang You 001 (Expert Syst. Appl., 2024)

Scalable Evidential K-Nearest Neighbor Classification on Big Data. - Chaoyu Gong; Jim Demmel; Yang You 001 (IEEE Trans. Big Data, 2024)

Distributed and Joint Evidential K-Nearest Neighbor Classification. - Chaoyu Gong; Jim Demmel; Yang You 001 (IEEE Trans. Knowl. Data Eng., 2024)

DavIR: Data Selection via Implicit Reward for Large Language Models. - Haotian Zhou; Tingkai Liu; Qianli Ma; Yufeng Zhang; Jianbo Yuan; Pengfei Liu; Yang You 001; Hongxia Yang (ACL (1), 2025)

Concerto: Automatic Communication Optimization and Scheduling for Large-Scale Deep Learning. - Shenggan Cheng; Shengjie Lin; Lansong Diao; Hao Wu 077; Siyu Wang; Chang Si; Ziming Liu; Xuanlei Zhao; Jiangsu Du; Wei Lin 016; Yang You 001 (ASPLOS (1), 2025)

Emphasizing Discriminative Features for Dataset Distillation in Complex Scenarios. - Kai Wang 036; Zekai Li; Zhi-Qi Cheng; Samir Khaki; Ahmad Sajedi; Ramakrishna Vedantam; Konstantinos N. Plataniotis; Alexander Hauptmann 001; Yang You 001 (CVPR, 2025)

A Closer Look at Time Steps is Worthy of Triple Speed-Up for Diffusion Model Training. - Kai Wang 036; Mingjia Shi; Yukun Zhou; Zekai Li; Zhihang Yuan; Yuzhang Shang; Xiaojiang Peng; Hanwang Zhang; Yang You 001 (CVPR, 2025)

A Stitch in Time Saves Nine: Small VLM is a Precise Guidance for Accelerating Large VLMs. - Wangbo Zhao; Yizeng Han; Jiasheng Tang; Zhikai Li; Yibing Song; Kai Wang 036; Zhangyang Wang; Yang You 001 (CVPR, 2025)

MixEval-X: Any-to-any Evaluations from Real-world Data Mixture. - Jinjie Ni; Yifan Song; Deepanway Ghosal; Bo Li; David Junhao Zhang; Xiang Yue; Fuzhao Xue; Yuntian Deng; Zian Zheng 001; Kaichen Zhang; Mahir Shah; Kabir Jain; et al. (ICLR, 2025)

Dynamic Diffusion Transformer. - Wangbo Zhao; Yizeng Han; Jiasheng Tang; Kai Wang 036; Yibing Song; Gao Huang 001; Fan Wang 019; Yang You 001 (ICLR, 2025)

Real-Time Video Generation with Pyramid Attention Broadcast. - Xuanlei Zhao; Xiaolong Jin; Kai Wang 036; Yang You 001 (ICLR, 2025)

WeiPipe: Weight Pipeline Parallelism for Communication-Effective Long-Context Large Model Training. - Junfeng Lin; Ziming Liu; Yang You 001; Jun Wang 159; Weihao Zhang; Rong Zhao (PPoPP, 2025)

Recurrent Diffusion for Large-Scale Parameter Generation. - Kai Wang 036; Dongwen Tang; Wangbo Zhao; Yang You 001 (CoRR, 2025)

CDIO: Cross-Domain Inference Optimization with Resource Preference Prediction for Edge-Cloud Collaboration. - Zheming Yang; Wen Ji; Qi Guo; Dieli Hu 001; Chang Zhao; Xiaowei Li; Xuanlei Zhao; Yi Zhao; Chaoyu Gong; Yang You 001 (CoRR, 2025)

Enhance-A-Video: Better Generated Video for Free. - Yang Luo; Xuanlei Zhao; Mengzhao Chen; Kaipeng Zhang; Wenqi Shao; Kai Wang 036; Zhangyang Wang; Yang You 001 (CoRR, 2025)

Region-Adaptive Sampling for Diffusion Transformers. - Ziming Liu; Yifan Yang 004; Chengruidong Zhang; Yiqi Zhang; Lili Qiu; Yang You 001; Yuqing Yang 001 (CoRR, 2025)

KMT-PLL: K-Means Cross-Attention Transformer for Partial Label Learning. - Jinfu Fan; Linqing Huang; Chaoyu Gong; Yang You 001; Min Gan; Zhongjie Wang (IEEE Trans. Neural Networks Learn. Syst., 2025)

Open-Sora 2.0: Training a Commercial-Level Video Generation Model in $200k. - Xiangyu Peng; Zangwei Zheng; Chenhui Shen; Tom Young; Xinying Guo; Binluo Wang; Hang Xu; Hongxin Liu; Mingyan Jiang; Wenjun Li; Yuhui Wang; Anbang Ye; et al. (CoRR, 2025)

PEBench: A Fictitious Dataset to Benchmark Machine Unlearning for Multimodal Large Language Models. - Zhaopan Xu; Pengfei Zhou; Weidong Tang; Jiaxin Ai; Wangbo Zhao; Xiaojiang Peng; Kai Wang 036; Yang You 001; Wenqi Shao; Hongxun Yao; Kaipeng Zhang (CoRR, 2025)

Dynamic Vision Mamba. - Mengxuan Wu; Zekai Li; Zhiyuan Liang; Moyang Li; Xuanlei Zhao; Samir Khaki; Zheng Zhu; Xiaojiang Peng; Konstantinos N. Plataniotis; Kai Wang 036; Wangbo Zhao; Yang You 001 (CoRR, 2025)

MDK12-Bench: A Multi-Discipline Benchmark for Evaluating Reasoning in Multimodal Large Language Models. - Pengfei Zhou; Fanrui Zhang; Xiaopeng Peng 001; Zhaopan Xu; Jiaxin Ai; Yansheng Qiu; Chuanhao Li 001; Zhen Li 026; Ming Li; Yukang Feng; Jianwen Sun; Haoquan Zhang; et al. (CoRR, 2025)

DyDiT++: Dynamic Diffusion Transformers for Efficient Visual Generation. - Wangbo Zhao; Yizeng Han; Jiasheng Tang; Kai Wang 036; Hao Luo 004; Yibing Song; Gao Huang 001; Fan Wang 019; Yang You 001 (CoRR, 2025)

Unsupervised Learning for Class Distribution Mismatch. - Pan Du; Wangbo Zhao; Xinai Lu; Nian Liu; Zhikai Li; Chaoyu Gong; Suyun Zhao; Hong Chen 001; Cuiping Li; Kai Wang; Yang You 001 (CoRR, 2025)

DD-Ranking: Rethinking the Evaluation of Dataset Distillation. - Zekai Li; Xinhao Zhong; Samir Khaki; Zhiyuan Liang; Yuhao Zhou; Mingjia Shi; Ziqiao Wang; Xuanlei Zhao; Wangbo Zhao; Ziheng Qin; Mengxuan Wu; Pengfei Zhou; et al. (CoRR, 2025)

REPA Works Until It Doesn't: Early-Stopped, Holistic Alignment Supercharges Diffusion Training. - Ziqiao Wang; Wangbo Zhao; Yuhao Zhou; Zekai Li; Zhiyuan Liang; Mingjia Shi; Xuanlei Zhao; Pengfei Zhou; Kaipeng Zhang; Zhangyang Wang; Kai Wang 036; Yang You 001 (CoRR, 2025)

SRDiffusion: Accelerate Video Diffusion Inference via Sketching-Rendering Cooperation. - Shenggan Cheng; Yuanxin Wei; Lansong Diao; Yong Liu 020; Bujiao Chen; Lianghua Huang; Yu Liu 023; Wenyuan Yu; Jiangsu Du; Wei Lin 016; Yang You 001 (CoRR, 2025)

Drag-and-Drop LLMs: Zero-Shot Prompt-to-Weights. - Zhiyuan Liang; Dongwen Tang; Yuhao Zhou; Xuanlei Zhao; Mingjia Shi; Wangbo Zhao; Zekai Li; Peihao Wang; Konstantin Schürholt; Damian Borth; Michael M. Bronstein; Yang You 001; et al. (CoRR, 2025)

MoNE: Replacing Redundant Experts with Lightweight Novices for Structured Pruning of MoE. - Geng Zhang; Yuxuan Han; Yuxuan Lou; Wangbo Zhao; Yiqi Zhang; Yang You 001 (CoRR, 2025)

HelixPipe: Efficient Distributed Training of Long Sequence Transformers with Attention Parallel Pipeline Parallelism. - Geng Zhang; Shenggan Cheng; Xuanlei Zhao; Ziming Liu; Yang You 001 (CoRR, 2025)

Neural-Driven Image Editing. - Pengfei Zhou; Jie Xia; Xiaopeng Peng 001; Wangbo Zhao; Zilong Ye; Zekai Li; Suorong Yang; Jiadong Pan; Yuanxiang Chen; Ziqiao Wang; Kai Wang 036; Qian Zheng; et al. (CoRR, 2025)

ElasticMM: Efficient Multimodal LLMs Serving with Elastic Multimodal Parallelism. - Zedong Liu; Shenggan Cheng; Guangming Tan; Yang You 001; Dingwen Tao (CoRR, 2025)

MDK12-Bench: A Comprehensive Evaluation of Multimodal Large Language Models on Multidisciplinary Exams. - Pengfei Zhou; Xiaopeng Peng 001; Fanrui Zhang; Zhaopan Xu; Jiaxin Ai; Yansheng Qiu; Chuanhao Li 001; Zhen Li 026; Ming Li; Yukang Feng; Jianwen Sun; Haoquan Zhang; et al. (CoRR, 2025)