Shitij Govil

about

Hi! I'm a final-year Computer Science undergraduate at Georgia Tech, where I work with Prof. Animesh Garg and Prof. Pan Li.

My research focuses on robot learning: building world-action models that learn transferable representations of motion and dynamics from large-scale human video. I am especially interested in using 3D scene flow and pretrained video-generation backbones to build sample-efficient robot policies that generalize across tasks, objects, and embodiments. I will be applying to PhD programs for Fall 2027.

Outside of research, I love staying active and playing sports. I boulder and lift a lot, and occasionally play pickup ultimate frisbee and volleyball. I also listen to a lot of music (mostly techno) and DJ. Sometimes I also urbex.

research

FloMo architecture for jointly predicting 3D scene flow and robot actions

FloMo: 3D Scene Flow as a World-Action Model Intermediate for Learning from Human Video

Submitted to ICRA

Shitij Govil*, James A. Collins*, Naman Patel*, Ayush Agarwal*, Animesh Garg

A 5B world-action model that predicts 3D scene flow and robot actions, using action-free human video to improve sample efficiency and generalization to unseen objects, motions, and tasks.

* Equal contribution

HEPTv2 single-stage end-to-end model and joint supervision from Figure 1

HEPTv2: End-to-End Efficient Point Transformer for Charged Particle Reconstruction

Under review at PRX Intelligence, 2026

Siqi Miao*, Shitij Govil*, Jack P. Rodgers, Mia Liu, Javier Duarte, Shih-Chieh Hsu, Yuan-Tang Chou, Pan Li

An end-to-end point transformer that reconstructs particle tracks without graph construction, clustering, or filtering, reaching 98.6% tracking efficiency with approximately 15 ms inference per event.

* Equal contribution

NeurIPS 2025 poster for the locality-sensitive hashing point transformer

Locality-Sensitive Hashing-Based Efficient Point Transformer for Charged Particle Reconstruction

Machine Learning and the Physical Sciences Workshop, NeurIPS 2025

Shitij Govil, Jack P. Rodgers, Yuan-Tang Chou, Siqi Miao, Amit Saha, Advaith Anand, Kilian Lieret, Gage DeZoort, Mia Liu, Javier Duarte, Pan Li, Shih-Chieh Hsu

Is Human-Written Data Enough? thumbnail

Is Human-Written Data Enough? The Challenge of Teaching Reasoning to LLMs Without RL or Distillation

AI for Math Workshop, ICML 2025

Wei Du, Branislav Kisacanin, George Armstrong, Shubham Toshniwal, Ivan Moshkov, Alexan Ayrapetyan, Sadegh Mahdavi, Dan Zhao, Shizhe Diao, Dragan Masulovic, Marius Stanean, Advaith Avadhanam, Max Wang, Ashmit Dutta, Shitij Govil, Sri Yanamandara, Mihir Tandon, Sriram Ananthakrishnan, Vedant Rathi, David Zhang, Joonseok Kang, Leon Luo, Titu Andreescu, Boris Ginsburg, Igor Gitman

Developed strong reasoning capabilities in base models without reinforcement learning via high-quality CoT.

Pre-training graph neural networks with structural fingerprints for materials discovery thumbnail

Pre-training graph neural networks with structural fingerprints for materials discovery

arXiv, 2025 [In Submission: Machine Learning Science and Technology]

Shuyi Jia, Shitij Govil, Manav Ramaprasad, Victor Fung

Pre-training GNNs for materials science using cheaply-computed structural fingerprints.

Using Hyperspatial LiDAR and Multispectral Imaging to Identify Coastal Wetlands thumbnail

Using Hyperspatial LiDAR and Multispectral Imaging to Identify Coastal Wetlands Using Gradient Boosting Methods

Remote Sensing, 2022

Shitij Govil, Aidan J. Lee, Aiden C. MacQueen, Narcisa G. Pricope, Asami Minei, Cuixian Chen

Developed a novel approach combining hyperspatial LiDAR and multispectral imaging data with gradient boosting methods to accurately identify and classify coastal wetlands.

projects

Efficient Skill-based Reinforcement Learning thumbnail

Efficient Skill-based Reinforcement Learning

Course Project CS 8803 DRL

A model-based RL framework that extracts reusable skills from rewardless offline data and reuses the data by relabeling with an optimistic reward estimator for efficient exploration through the learned dynamics model.

Winner-Take-All Sparse Autoencoders thumbnail

Winner-Take-All Sparse Autoencoders

Course Project CS 7461 DL

A winner-take-all sparsity mechanism for sparse autoencoders in the context of mechanistic interpretability. Investigates feature-wise sparsity constraints for SAEs to prevent dead features.