English
Related papers

Related papers: Efficient Training of Transformers for Molecule Pr…

200 papers

Driver vigilance estimation is an important task for transportation safety. Wearable and portable brain-computer interface devices provide a powerful means for real-time monitoring of the vigilance level of drivers to help with avoiding…

Machine Learning · Computer Science 2021-06-15 Guangyi Zhang , Ali Etemad

Attention mechanisms have raised significant interest in the research community, since they promise significant improvements in the performance of neural network architectures. However, in any specific problem, we still lack a principled…

Computer Vision and Pattern Recognition · Computer Science 2021-12-24 Rafael Pedro , Arlindo L. Oliveira

Attention-based neural networks have achieved state-of-the-art results on a wide range of tasks. Most such models use deterministic attention while stochastic attention is less explored due to the optimization difficulties or complicated…

Machine Learning · Computer Science 2021-06-10 Shujian Zhang , Xinjie Fan , Bo Chen , Mingyuan Zhou

Transformer has achieved great success in NLP. However, the quadratic complexity of the self-attention mechanism in Transformer makes it inefficient in handling long sequences. Many existing works explore to accelerate Transformers by…

Computation and Language · Computer Science 2021-09-03 Chuhan Wu , Fangzhao Wu , Tao Qi , Binxing Jiao , Daxin Jiang , Yongfeng Huang , Xing Xie

Transformers often appear to perform Bayesian reasoning in context, but verifying this rigorously has been impossible: natural data lack analytic posteriors, and large models conflate reasoning with memorization. We address this by…

Machine Learning · Computer Science 2026-05-19 Naman Agarwal , Siddhartha R. Dalal , Vishal Misra

Deep imitation learning is promising for solving dexterous manipulation tasks because it does not require an environment model and pre-programmed robot behavior. However, its application to dual-arm manipulation tasks remains challenging.…

Robotics · Computer Science 2025-05-23 Heecheol Kim , Yoshiyuki Ohmura , Yasuo Kuniyoshi

Accurate quantification of cerebral blood flow (CBF) is essential for the diagnosis and assessment of a wide range of neurological diseases. Positron emission tomography (PET) with radiolabeled water (15O-water) is considered the…

Computer Vision and Pattern Recognition · Computer Science 2022-11-23 Ramy Hussein , David Shin , Moss Zhao , Jia Guo , Guido Davidzon , Michael Moseley , Greg Zaharchuk

The Transformer architecture has revolutionized deep learning, delivering the state-of-the-art performance in areas such as natural language processing, computer vision, and time series prediction. However, its core component,…

Machine Learning · Computer Science 2025-07-24 Haolin Jin , Mengbai Xiao , Yuan Yuan , Xiao Zhang , Dongxiao Yu , Guanghui Zhang , Haoliang Wang

We introduce the Block Transformer which adopts hierarchical global-to-local modeling to autoregressive transformers to mitigate the inference bottlenecks associated with self-attention. Self-attention requires the key-value (KV) cache of…

Computation and Language · Computer Science 2024-11-04 Namgyu Ho , Sangmin Bae , Taehyeon Kim , Hyunjik Jo , Yireun Kim , Tal Schuster , Adam Fisch , James Thorne , Se-Young Yun

Transformers are slow and memory-hungry on long sequences, since the time and memory complexity of self-attention are quadratic in sequence length. Approximate attention methods have attempted to address this problem by trading off model…

Machine Learning · Computer Science 2022-06-24 Tri Dao , Daniel Y. Fu , Stefano Ermon , Atri Rudra , Christopher Ré

We propose a novel approach for predicting protein-peptide interactions using a bi-modal transformer architecture that learns an inter-facial joint distribution of residual contacts. The current data sets for crystallized protein-peptide…

Biomolecules · Quantitative Biology 2023-06-02 Justin Diamond , Markus Lill

Diffusion models have recently gained state of the art performance on many image generation tasks. However, most models require significant computational resources to achieve this. This becomes apparent in the application of medical image…

Computer Vision and Pattern Recognition · Computer Science 2025-04-16 Karan Jain , Mohammad Nayeem Teli

Focused ultrasound (FUS) can be used to open the blood-brain barrier (BBB), and MRI with contrast agents can detect that opening. However, repeated use of gadolinium-based contrast agents (GBCAs) presents safety concerns to patients. This…

Quantitative Methods · Quantitative Biology 2023-01-19 P. Lee , H. Wei , A. N. Pouliopoulos , B. T. Forsyth , Y. Yang , C. Zhang , A. F. Laine , E. E. Konofagou , C. Wu , J. Guo

Attention mechanism has been extensively integrated within mainstream neural network architectures, such as Transformers and graph attention networks. Yet, its underlying working principles remain somewhat elusive. What is its essence? Are…

Machine Learning · Computer Science 2024-12-25 Tianyu Ruan , Shihua Zhang

In this paper, we introduce the prior knowledge, multi-scale structure, into self-attention modules. We propose a Multi-Scale Transformer which uses multi-scale multi-head self-attention to capture features from different scales. Based on…

Computation and Language · Computer Science 2019-12-03 Qipeng Guo , Xipeng Qiu , Pengfei Liu , Xiangyang Xue , Zheng Zhang

Novel approaches for predicting the outcomes of blood-brain barrier (BBB) opening with focused ultrasound (FUS) and microbubbles are highly desired. This study aims to explore machine learning-based methods for reliably predicting the…

Medical Physics · Physics 2023-06-08 Wenjing Li , Chenchen Bing , Haixin Dai , Rajiv Chopra , Qian Wang , Bingbing Cheng

Skin lesion segmentation from dermoscopy images is of great importance for improving the quantitative analysis of skin cancer. However, the automatic segmentation of melanoma is a very challenging task owing to the large variation of…

Image and Video Processing · Electrical Eng. & Systems 2021-10-11 Jiacheng Wang , Lan Wei , Liansheng Wang , Qichao Zhou , Lei Zhu , Jing Qin

Prostate cancer biopsy benefits from accurate fusion of transrectal ultrasound (TRUS) and magnetic resonance (MR) images. In the past few years, convolutional neural networks (CNNs) have been proved powerful in extracting image features…

Computer Vision and Pattern Recognition · Computer Science 2021-07-13 Xinrui Song , Hengtao Guo , Xuanang Xu , Hanqing Chao , Sheng Xu , Baris Turkbey , Bradford J. Wood , Ge Wang , Pingkun Yan

Beamforming has significance for enhancing spectral efficiency and mitigating interference in multi-antenna wireless systems, facilitating spatial multiplexing and diversity in dense and high mobility scenarios. Traditional beamforming…

Information Theory · Computer Science 2025-11-06 Cemil Vahapoglu , Timothy J. O'Shea , Wan Liu , Sennur Ulukus

With the upcoming large-scale surveys like LSST, we expect to find approximately $10^5$ strong gravitational lenses among data of many orders of magnitude larger. In this scenario, the usage of non-automated techniques is too time-consuming…

Astrophysics of Galaxies · Physics 2024-03-20 Hareesh Thuruthipilly , Margherita Grespan , Adam Zadrożny