← Search

Tong Geng

14 accepted papers

2026

Zeros can be Informative: Masked Binary U-Net for Image Segmentation on Tensor Cores

ICLR 2026poster

Real-time image segmentation is a key enabler for AR/VR, robotics, drones, and autonomous systems, where tight accuracy, latency, and energy budgets must be met on resource‑constrained edge devices. While U‑Net offers a favorable balance of accuracy and efficiency compared to large transformer‑based…

Cited by 0SourcecodeScholar
2025

An Expressive and Self-Adaptive Dynamical System for Efficient Function Learning

ICML 2025poster

Function learning forms the foundation of numerous scientific and engineering tasks. While modern machine learning (ML) methods model complex functions effectively, their escalating complexity and computational demands pose challenges to efficient deployment. In contrast, natural dynamical systems e…

Cited by 0SourcePDFScholar
2025

DS-LLM: Leveraging Dynamical Systems to Enhance Both Training and Inference of Large Language Models

ICLR 2025poster

The training of large language models (LLMs) faces significant computational cost challenges, limiting their scalability toward artificial general intelligence (AGI) and broader adoption. With model sizes doubling approximately every 3.4 months and training costs escalating from 64 million USD for G…

Cited by 2SourcePDFScholar
2025

Diff-PIC: Revolutionizing Particle-In-Cell Nuclear Fusion Simulation with Diffusion Models

ICLR 2025poster

The rapid development of AI highlights the pressing need for sustainable energy, a critical global challenge for decades. Nuclear fusion, generally seen as a promising solution, has been the focus of intensive research for nearly a century, with investments reaching hundreds of billions of dollars.…

Cited by 0SourcePDFScholar
2025

InstaTrain: Adaptive Training via Ultra-Fast Natural Annealing within Dynamical Systems

ICLR 2025poster

Time-series modeling is broadly adopted to capture underlying patterns present in historical data, allowing prediction of future values. However, one crucial aspect of such modeling is often overlooked: in highly dynamic environments, data distributions can shift drastically within a second or less.…

Cited by 1SourcePDFScholar
2025

MEPT: Mixture of Expert Prompt Tuning as a Manifold Mapper

EMNLP 2025

Considering deep neural networks as manifold mappers, the pretrain-then-fine-tune paradigm can be interpreted as a two-stage process: pretrain establishes a broad knowledge base, and fine-tune adjusts the model parameters to activate specific neural pathways to align with the target manifold. Althou

Cited by 0SourcePDFScholar
2025

Probabilistic Token Alignment for Large Language Model Fusion

NeurIPS 2025poster

Training large language models (LLMs) from scratch can yield models with unique functionalities and strengths, but it is costly and often leads to redundant capabilities. A more cost-effective alternative is to fuse existing pre-trained LLMs with different architectures into a more powerful model. H…

Cited by 0SourceScholar
2025

Visual Agents as Fast and Slow Thinkers

ICLR 2025poster

Achieving human-level intelligence requires refining cognitive distinctions between \textit{System 1} and \textit{System 2} thinking. While contemporary AI, driven by large language models, demonstrates human-like traits, it falls short of genuine cognition. Transitioning from structured benchmarks…

2024

Extending Power of Nature from Binary to Real-Valued Graph Learning in Real World

ICLR 2024poster

Nature performs complex computations constantly at clearly lower cost and higher performance than digital computers. It is crucial to understand how to harness the unique computational power of nature in Machine Learning (ML). In the past decade, besides the development of Neural Networks (NNs), the…

Cited by 9SourcePDFScholar
2024

Prototypical Transformer As Unified Motion Learners

ICML 2024poster

In this work, we introduce the Prototypical Transformer (ProtoFormer), a general and unified framework that approaches various motion tasks from a prototype perspective. ProtoFormer seamlessly integrates prototype learning with Transformer by thoughtfully considering motion dynamics, introducing two…

Cited by 17SourcePDFScholar
2023

ClusterFomer: Clustering As A Universal Visual Learner

NeurIPS 2023poster

This paper presents ClusterFormer, a universal vision model that is based on the Clustering paradigm with TransFormer. It comprises two novel designs: 1) recurrent cross-attention clustering, which reformulates the cross-attention mechanism in Transformer and enables recursive updates of cluster cen…

2023

LinGCN: Structural Linearized Graph Convolutional Network for Homomorphically Encrypted Inference

NeurIPS 2023poster

The growth of Graph Convolution Network (GCN) model sizes has revolutionized numerous applications, surpassing human performance in areas such as personal healthcare and financial systems. The deployment of GCNs in the cloud raises privacy concerns due to potential adversarial attacks on client dat…

2023

TransFlow: Transformer As Flow Learner

CVPR 2023highlight

Optical flow is an indispensable building block for various important computer vision tasks, including motion estimation, object tracking, and disparity measurement. In this work, we propose TransFlow, a pure transformer architecture for optical flow estimation. Compared to dominant CNN-based method…

Cited by 99SourcePDFScholar