Wei, Jia
- SpargeAttention: Accurate and Training-free Sparse Attention Accelerating Any Model Inference
2025/02/25 by Jintao Zhang, Zhang, Jintao, Chendong Xiang +11 · 53 citations
Computer Science · #Advanced Neural Network Applications #Artificial Intelligence (cs.AI) #Computer Vision and Pattern Recognition (cs.CV) #Domain Adaptation and Few-Shot Learning #FOS: Computer and information sciences #Machine Learning (cs.LG) #Performance (cs.PF) #Topic Modeling
- SageAttention: Accurate 8-Bit Attention for Plug-and-play Inference Acceleration
2024/10/03 by Jintao Zhang, Zhang, Jintao, Wei Jia +8 · 34 citations
Computer Science · #Anomaly Detection Techniques and Applications #FOS: Computer and information sciences #Machine Learning (cs.LG) #Seismology and Earthquake Studies
- SageAttention2: Efficient Attention with Thorough Outlier Smoothing and Per-thread INT4 Quantization
2024/11/17 by Jintao Zhang, Haofeng Huang, Zhang, Jintao +9 · 25 citations
Computer Science · Medicine · #Anomaly Detection Techniques and Applications #COVID-19 diagnosis using AI #Seismology and Earthquake Studies
- SageAttention3: Microscaling FP4 Attention for Inference and An Exploration of 8-Bit Training
2025/05/16 by Jintao Zhang, Zhang, Jintao, Wei Jia +15 · 14 citations
Computer Science · #Advanced Neural Network Applications #Artificial Intelligence (cs.AI) #Big Data and Digital Economy #Computer Vision and Pattern Recognition (cs.CV) #Domain Adaptation and Few-Shot Learning #FOS: Computer and information sciences #Hardware Architecture (cs.AR) #Machine Learning (cs.LG) #Performance (cs.PF)
- SageAttention2++: A More Efficient Implementation of SageAttention2
2025/05/27 by Jintao Zhang, Zhang, Jintao, Xiaoming Xu +13 · 7 citations
Computer Science · #Advanced Neural Network Applications #Artificial Intelligence (cs.AI) #Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences #Generative Adversarial Networks and Image Synthesis #Hardware Architecture (cs.AR) #Machine Learning (cs.LG) #Multimodal Machine Learning Applications
- Accurate INT8 Training Through Dynamic Block-Level Fallback
2025/03/11 by Pengle Zhang, Jia Wei, Zhang, Pengle +7 · 3 citations
Computer Science · #Advanced Neural Network Applications #FOS: Computer and information sciences #Generative Adversarial Networks and Image Synthesis #Machine Learning (cs.LG) #Parallel Computing and Optimization Techniques
- A Compact Neural Network-based Algorithm for Robust Image Watermarking
2021/12/27 by Xu, Hong-Bo, Wang, Rong, Wei, Jia +1 · 1 citation
#Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences
- Unsupervised Multi-Modal Medical Image Registration via Discriminator-Free Image-to-Image Translation
2022/04/28 by Chen, Zekang, Wei, Jia, Li, Rui · 1 citation
#Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences
- SFusion: Self-attention based N-to-One Multimodal Fusion Block
2022/08/26 by Zecheng Liu, Liu, Zecheng, Wei Jia +4 · 1 citation
Computer Science · Engineering · #Advanced Chemical Sensor Technologies #Artificial Intelligence (cs.AI) #Computer Vision and Pattern Recognition (cs.CV) #FOS: Computer and information sciences #Gaze Tracking and Assistive Technology #Hand Gesture Recognition Systems
- Dirigo: A Method to Extract Event Logs for Object-Centric Processes
2024/11/12 by Wei Jia, Wei, Jia, Chun Ouyang +7 · 1 citation
Decision Sciences · Computer Science · #Scientific Computing and Data Management #Advanced Database Systems and Queries
- When FinTech Meets Privacy: Securing Financial LLMs with Differential Private Fine-Tuning
2025/09/10 by Zhu, Sichen, Leung, Hoyeung, Wang, Xiaoyi +2 · 1 citation
#Cryptography and Security (cs.CR) #FOS: Computer and information sciences