English
Related papers

Related papers: Solvaformer: an SE(3)-equivariant graph transforme…

200 papers

Transformer-based models have recently shown success in representation learning on graph-structured data beyond natural language processing and computer vision. However, the success is limited to small-scale graphs due to the drawbacks of…

Machine Learning · Computer Science 2022-10-05 Jinyoung Park , Seongjun Yun , Hyeonjin Park , Jaewoo Kang , Jisu Jeong , Kyung-Min Kim , Jung-woo Ha , Hyunwoo J. Kim

We present a SE(3)-equivariant graph neural network (GNN) approach that directly predicting the formation factor and effective permeability from micro-CT images. FFT solvers are established to compute both the formation factor and effective…

Exploiting relations among 2D joints plays a crucial role yet remains semi-developed in 2D-to-3D pose estimation. To alleviate this issue, we propose GraFormer, a novel transformer architecture combined with graph convolution for 3D pose…

Computer Vision and Pattern Recognition · Computer Science 2021-09-20 Weixi Zhao , Yunjie Tian , Qixiang Ye , Jianbin Jiao , Weiqiang Wang

Features that are equivariant to a larger group of symmetries have been shown to be more discriminative and powerful in recent studies. However, higher-order equivariant features often come with an exponentially-growing computational cost.…

Computer Vision and Pattern Recognition · Computer Science 2021-04-05 Haiwei Chen , Shichen Liu , Weikai Chen , Hao Li

Spectral Graph Neural Networks have demonstrated superior performance in graph representation learning. However, many current methods focus on employing shared polynomial coefficients for all nodes, i.e., learning node-unified filters,…

Machine Learning · Computer Science 2024-07-22 Jiahong Ma , Mingguo He , Zhewei Wei

Transformers have become methods of choice in many applications thanks to their ability to represent complex interactions between elements. However, extending the Transformer architecture to non-sequential data such as molecules and…

Machine Learning · Computer Science 2022-04-27 Yoni Choukroun , Lior Wolf

Directed graphs are widely used to model asymmetric relationships in real-world systems. However, existing directed graph neural networks often struggle to jointly capture directional semantics and global structural patterns due to their…

Machine Learning · Computer Science 2025-08-20 Jiayu Fang , Zhiqi Shao , S T Boris Choy , Junbin Gao

Robust and efficient interpretation of QSAR methods is quite useful to validate AI prediction rationales with subjective opinion (chemist or biologist expertise), understand sophisticated chemical or biological process mechanisms, and…

Biomolecules · Quantitative Biology 2026-05-05 Jinjiang Guo , Qi Liu , Han Guo , Xi Lu

Deep learning has emerged as a transformative tool for the neural surrogate modeling of partial differential equations (PDEs), known as neural PDE solvers. However, scaling these solvers to industrial-scale geometries with over $10^8$ cells…

Machine Learning · Computer Science 2026-02-06 Hang Zhou , Haixu Wu , Haonan Shangguan , Yuezhou Ma , Huikun Weng , Jianmin Wang , Mingsheng Long

Group equivariance (e.g. SE(3) equivariance) is a critical physical symmetry in science, from classical and quantum physics to computational biology. It enables robust and accurate prediction under arbitrary reference transformations. In…

Computational Engineering, Finance, and Science · Computer Science 2023-02-08 Weitao Du , He Zhang , Yuanqi Du , Qi Meng , Wei Chen , Bin Shao , Tie-Yan Liu

In recent years, molecular representation learning has emerged as a key area of focus in various chemical tasks. However, many existing models fail to fully consider the geometric information of molecular structures, resulting in less…

Machine Learning · Computer Science 2023-06-29 Bumju Kwak , Jiwon Park , Taewon Kang , Jeonghee Jo , Byunghan Lee , Sungroh Yoon

Accurate forecasting of solar power output is essential for efficient integration of renewable energy into the grid. In this study, an attention-based deep learning model, inspired by transformer architecture, is used for short-term solar…

Machine Learning · Computer Science 2026-04-28 Ankan Basu , Jyotiraditya Roy , Aditya Datta , Prayas Sanyal , Sumanta Banerjee

Traffic forecasting is a cornerstone of smart city management, enabling efficient resource allocation and transportation planning. Deep learning, with its ability to capture complex nonlinear patterns in spatiotemporal (ST) data, has…

Machine Learning · Computer Science 2024-10-16 Hongjun Wang , Jiyuan Chen , Tong Pan , Zheng Dong , Lingyu Zhang , Renhe Jiang , Xuan Song

Molecular 3D conformations play a key role in determining how molecules interact with other molecules or protein surfaces. Recent deep learning advancements have improved conformation prediction, but slow training speeds and difficulties in…

Machine Learning · Computer Science 2024-12-17 Qingwen Tian , Yuxin Xu , Yixuan Yang , Zhen Wang , Ziqi Liu , Pengju Yan , Xiaolin Li

Signed Graph Neural Networks (SGNNs) have been shown to be effective in analyzing complex patterns in real-world situations where positive and negative links coexist. However, SGNN models suffer from poor explainability, which limit their…

Machine Learning · Computer Science 2024-12-13 Lu Li , Jiale Liu , Xingyu Ji , Maojun Wang , Zeyu Zhang

Graph Transformer has demonstrated impressive capabilities in the field of graph representation learning. However, existing approaches face two critical challenges: (1) most models suffer from exponentially increasing computational…

Effectively preserving and encoding structure features from objects in irregular and sparse LiDAR points is a key challenge to 3D object detection on point cloud. Recently, Transformer has demonstrated promising performance on many 2D and…

Computer Vision and Pattern Recognition · Computer Science 2022-09-07 Xiaoyu Feng , Heming Du , Yueqi Duan , Yongpan Liu , Hehe Fan

This paper introduces PDEformer, a neural solver for partial differential equations (PDEs) capable of simultaneously addressing various types of PDEs. We propose to represent the PDE in the form of a computational graph, facilitating the…

Numerical Analysis · Mathematics 2025-01-28 Zhanhong Ye , Xiang Huang , Leheng Chen , Hongsheng Liu , Zidong Wang , Bin Dong

Learning representations on large graphs is a long-standing challenge due to the inter-dependence nature. Transformers recently have shown promising performance on small graphs thanks to its global attention for capturing all-pair…

Machine Learning · Computer Science 2024-09-16 Qitian Wu , Kai Yang , Hengrui Zhang , David Wipf , Junchi Yan

Accurate predictions of interatomic energies and forces are essential for high quality molecular dynamic simulations (MD). Machine learning algorithms can be used to overcome limitations of classical MD by predicting ab initio quality…

Chemical Physics · Physics 2022-01-04 Bryce Hedelius , Fabian B. Fuchs , Dennis Della Corte