Large Language Models (LLMs) exhibit exceptional proficiency in handling extensive context windows in natural language. Nevertheless, the quadratic scaling of attention computation relative to sequence length creates substantial efficiency bottlenecks, necessitating the development of I/O-optimized algorithms. In this work, we conduct a systematic examination of the I/O complexity inherent in attention mechanisms, with a specific emphasis on the backward pass under both small and large cache settings. By leveraging the red-blue pebble game framework, we derive tight bounds for I/O complexity across the full spectrum of cache sizes. We validate that FlashAttention, one of the current industry standards, achieves optimality in the large-cache scenario for both forward and backward passes. Conversely, for small-cache environments, we introduce a novel algorithm that outperforms contemporary methods and successfully attains theoretical tight bounds. Furthermore, we expand our investigation to include sparse attention by establishing granular lower bounds for both forward and backward passes across all cache configurations. Ultimately, our results solidify the theoretical framework regarding I/O complexity in attention mechanisms, providing critical guidance for the development of efficient LLM training and inference systems.
Mind the Generation Process: Fine-Grained Confidence Estimation During LLM Generation
Shanghai Institute of Artificial Intelligence for Education(上海人工智能教育研究院);East China Normal University(华东师范大学);School of Data Science, Fudan University(复旦大学数据科学学院);College of Computer Science and Artificial Intelligence, Fudan University(复旦大学计算机科学与人工智能学院);Alibaba(阿里巴巴集团)
Outcome-Grounded Advantage Reshaping for Fine-Grained Credit Assignment in Mathematical Reasoning
Fudan University(复旦大学);XingYun lab, HUJING Digital Media & Entertainment Group(星云实验室,HUJING数字媒体与娱乐集团);University of Science and Technology Beijing(北京科技大学);Chinese Academy of Sciences(中国科学院);Beijing University of Posts and Telecommunications(北京邮电大学)
BAS-OPD: Budget-Aware Selective On-Policy Self-Distillation for Fine-Grained Multimodal Perception
HFIPS, Chinese Academy of Sciences(中国科学院合肥物质科学研究院);University of Science and Technology of China(中国科学技术大学);University of California, Los Angeles(加利福尼亚大学洛杉矶分校)
Shanghai Research Institute of ETstar information tech. co. ltd(上海星途信息技术有限公司上海研究院)
We present a work-flow which aims at capturing residents' abnormal activities through the passenger flow of elevator in multi-storey residence buildings. Camera and sensors (hall sensor, photoelectric sensor, gyro, accelerometer, barometer, and thermometer) with internet connection are mounted in elevator to collect image and data. Computer vision algorithms such as instance segmentation, multi-label recognition, embedding and clustering are applied to generalize passenger flow of elevator, i.e. how many people and what kinds of people get in and out of the elevator on each floor. More specifi…
Amazon.com Inc(亚马逊公司);ByteDance Inc(字节跳动公司);University of Pittsburgh(匹兹堡大学);Snap Inc(Snap公司)
Deep learning models in large-scale machine learning systems are often continuously trained with enormous data from production environments. The sheer volume of streaming training data poses a significant challenge to real-time training subsystems and ad-hoc sampling is the standard practice. Our key insight is that these deployed ML systems continuously perform forward passes on data instances during inference, but ad-hoc sampling does not take advantage of this substantial computational effort. Therefore, we propose to record a constant amount of information per instance from these forward p…
FoodSky: A Food-oriented Large Language Model that Passes the Chef and Dietetic Examination
Institute of Computing Technology, Chinese Academy of Sciences(中国科学院计算技术研究所);University of Chinese Academy of Sciences(中国科学院大学);Institute of Intelligent Computing Technology, Chinese Academy of Sciences(中国科学院智能计算技术研究所);Zhongke Shenjian (Suzhou) Technology Co., Ltd(中科神箭(苏州)科技有限公司)
Reducing Fine-Tuning Memory Overhead by Approximate and Memory-Sharing Backpropagation
School of Statistics and Data Science, Nankai University(南开大学统计与数据科学学院);School of Information Science and Technology, ShanghaiTech University(上海科技大学信息科学与技术学院);Department of Automation, Tsinghua University(清华大学自动化系);Central Research Institute, United Imaging Healthcare, Co., Ltd.(联影医疗股份有限公司中央研究院)
Backward Lens: Projecting Language Model Gradients into the Vocabulary Space
Faculty of Computer Science, Technion – Israel Institute of Technology(以色列理工学院计算机科学系);Blavatnik School of Computer Science, Tel Aviv University(特拉维夫大学布拉瓦尼克计算机科学学院)
Machine Learning Group, Technische Universität Berlin(柏林技术大学机器学习组);BIFOLD – Berlin Institute for the Foundations of Learning and Data(柏林学习与数据基础研究院)
Activation patching is a standard method in mechanistic interpretability for localizing the components of a model responsible for specific behaviors, but it is computationally expensive to apply at scale. Attribution patching offers a faster, gradient-based approximation, yet suffers from noise and reduced reliability in deep, highly non-linear networks. In this work, we introduce Relevance Patching (RelP), which replaces the local gradients in attribution patching with propagation coefficients derived from Layer-wise Relevance Propagation (LRP). LRP propagates the network's output backward th…
University of Toronto Scarborough(多伦多大学士嘉堡校区);University of Toronto(多伦多大学);University of Pennsylvania(宾夕法尼亚大学);Mila(米拉研究所);McGill University(麦吉尔大学);Canadian Institute for Advanced Research(加拿大高级研究所)
In artificial neural networks trained with gradient descent, the weights used for processing stimuli are also used during backward passes to calculate gradients. For the real brain to approximate gradients, gradient information would have to be propagated separately, such that one set of synaptic weights is used for processing and another set is used for backward passes. This produces the so-called "weight transport problem" for biological models of learning, where the backward weights used to calculate gradients need to mirror the forward weights used to process stimuli. This weight transport…
The aim of this paper is to introduce a new learning procedure for neural networks and to demonstrate that it works well enough on a few small problems to be worth further investigation. The Forward-Forward algorithm replaces the forward and backward passes of backpropagation by two forward passes, one with positive (i.e. real) data and the other with negative data which could be generated by the network itself. Each layer has its own objective function which is simply to have high goodness for positive data and low goodness for negative data. The sum of the squared activities in a layer can b…
FC Barcelona(巴塞罗那足球俱乐部);Polytechnic University of Catalonia(加泰罗尼亚理工大学);Simon Fraser University(西蒙弗雷泽大学);Zelus Analytics(泽卢斯分析公司)
The expected possession value (EPV) of a soccer possession represents the likelihood of a team scoring or receiving the next goal at any time instance. By decomposing the EPV into a series of subcomponents that are estimated separately, we develop a comprehensive analysis framework providing soccer practitioners with the ability to evaluate the impact of both observed and potential actions. We show we can obtain calibrated models for all the components of EPV, including a set of yet-unexplored problems in soccer. We produce visually-interpretable probability surfaces for potential passes from …
Backward-State Policy Is Part of the Learning Algorithm
Beijing Tongming Lake Information Technology Application Innovation Center (TLAIC)(北京通明湖信息技术应用创新中心);Fudan University Institute of Systems for Advanced Computing(复旦大学先进计算系统研究所);Harbin Institute of Technology(哈尔滨工业大学);Key Lab of HCST (PKU), MOE(北京大学高可信软件技术教育部重点实验室;北京大学计算机学院);SCS, Peking University(上海开放计算系统研究所);Shanghai Institute of Systems for Open Computing
Moravian Academy(摩拉维亚学院);Lehigh University(理海大学);MGH & BWH Center for Clinical Data Science(麻省总医院与布莱根妇女医院临床数据科学中心)
Although machine learning has become a powerful tool to augment doctors in clinical analysis, the immense amount of labeled data that is necessary to train supervised learning approaches burdens each development task as time and resource intensive. The vast majority of dense clinical information is stored in written reports, detailing pertinent patient information. The challenge with utilizing natural language data for standard model development is due to the complex nature of the modality. In this research, a model pipeline was developed to utilize an unsupervised approach to train an encoder…
SeFAR: Semi-supervised Fine-grained Action Recognition with Temporal Perturbation and Learning Stabilization
Unmanned System Research Institute, Northwestern Polytechnical University(西北工业大学无人系统研究院);School of Automation, Northwestern Polytechnical University(西北工业大学自动化学院);School of Computer Science, Northwestern Polytechnical University(西北工业大学计算机学院);School of Software, Northwestern Polytechnical University(西北工业大学软件学院)