Selected research
Research Contributions
My prior research through the Rensselaer-IBM AI Research Collaboration focused on graph transformer architectures for structural graph representations, molecular geometry, and efficient attention.
KDD 2022
Graph transformers
Edge-augmented Graph Transformer (EGT)
EGT is an early graph transformer architecture that introduced evolving structural edge channels for joint node and pair representations across graph-level prediction, edge classification, and link prediction.
Author and maintainer of the PyTorch and TensorFlow implementations.
KDD 2023
Efficient attention
Stochastically Subsampled Self-Attention (SSA)
SSA subsamples transformer self-attention during training and formulates the Information Pathways Hypothesis, reducing attention memory and computation by 4–8× across graph, vision, and language tasks.
ICML 2024
Molecular geometry
Triplet Graph Transformer (TGT)
TGT extends EGT with third-order interactions among groups of three atoms and a 2D-to-3D framework that predicts interatomic distances from 2D molecular graphs for property prediction.
Author and maintainer of the official PyTorch implementation.
TGT has remained the top-ranked non-ensemble submission on PCQM4Mv2 for more than two years and records the highest EwT among direct methods on the Open Catalyst 2020 (OC20) IS2RE leaderboard.