← Search

Weian Mao

10 accepted papers

2026

TriAttention: Efficient Long Reasoning with Trigonometric KV Compression

ICML 2026poster

Extended reasoning in large language models (LLMs) requires long and accurate decoding and creates severe KV cache memory bottlenecks. Leading KV cache compression methods estimate KV importance using attention scores from recent post-RoPE queries. However, queries rotate with position during RoPE, …

Cited by 0SourceScholar
2025

Boltzmann-Aligned Inverse Folding Model as a Predictor of Mutational Effects on Protein-Protein Interactions

ICLR 2025spotlight

Predicting the change in binding free energy ($\Delta \Delta G$) is crucial for understanding and modulating protein-protein interactions, which are critical in drug design. Due to the scarcity of experimental $\Delta\Delta G$ data, existing methods focus on pre-training, while neglecting the impo…

2025

Revisiting Convolution Architecture in the Realm of DNA Foundation Models

ICLR 2025poster

In recent years, A variety of methods based on Transformer and state space model (SSM) architectures have been proposed, advancing foundational DNA language models. However, there is a lack of comparison between these recent approaches and the classical architecture—convolutional networks (CNNs)—on…

Cited by 0SourcePDFScholar
2024

De novo Protein Design Using Geometric Vector Field Networks

ICLR 2024spotlight

Advances like protein diffusion have marked revolutionary progress in $\textit{de novo}$ protein design, a central topic in life science. These methods typically depend on protein structure encoders to model residue backbone frames, where atoms do not exist. Most prior encoders rely on atom-wise fea…

2024

Floating Anchor Diffusion Model for Multi-motif Scaffolding

ICML 2024poster

Motif scaffolding seeks to design scaffold structures for constructing proteins with functions derived from the desired motif, which is crucial for the design of vaccines and enzymes. Previous works approach the problem by inpainting or conditional generation. Both of them can only scaffold motifs w…

2024

Generative Active Learning for Long-tailed Instance Segmentation

ICML 2024poster

Recently, large-scale language-image generative models have gained widespread attention and many works have utilized generated data from these models to further enhance the performance of perception tasks. However, not all generated data can positively impact downstream models, and these methods do…

2023

CTVIS: Consistent Training for Online Video Instance Segmentation

ICCV 2023poster

The discrimination of instance embeddings plays a vital role in associating instances across time for online video instance segmentation (VIS). Instance embedding learning is directly supervised by the contrastive loss computed upon the contrastive items (CIs), which are sets of anchor/positive/nega…

Cited by 46PDFcodeScholar
2023

SegPrompt: Boosting Open-World Segmentation via Category-Level Prompt Learning

ICCV 2023poster

Current closed-set instance segmentation models rely on predefined class labels for each mask during training and evaluation, limiting their ability to detect novel objects. Open-world instance segmentation (OWIS) models address this challenge by detecting unknown objects in a class-agnostic manner.…

Cited by 21PDFcodeScholar
2022

Poseur: Direct Human Pose Regression with Transformers

ECCV 2022poster

"We propose a direct, regression-based approach to 2D human pose estimation from single images. We formulate the problem as a sequence prediction task, which we solve using a Transformer network. This network directly learns a regression mapping from images to the keypoint coordinates, without resor…

2021

FCPose: Fully Convolutional Multi-Person Pose Estimation With Dynamic Instance-Aware Convolutions

CVPR 2021poster

We propose a fully convolutional multi-person pose estimation framework using dynamic instance-aware convolutions, termed FCPose. Different from existing methods, which often require ROI (Region of Interest) operations and/or grouping post-processing, FCPose eliminates the ROIs and grouping post-pro…

Cited by 81PDFScholar