PyTorch Geometric Contribution: COFT

Dec. 2025 | CS 224W, Machine Learning with Graphs [with Hsin (Newton) Chen]

Contributed a full implementation of COFT (Coarse-to-Fine Highlighting) into the PyTorch Geometric (PyG) framework, enabling structured entity-aware highlighting for hallucination reduction in LLM long-context reasoning tasks. The module includes:

• ⚡ Recaller with Wikidata entity expansion
• 📊 Scorer with TF-ISF + self-information weighting
• 🧠 Selector with dynamic thresholding and flexible granularity
• 🧪 Full test suite & examples integrated into PyG

Our project also includes a Medium deep-dive article explaining our approach, architecture, and integration challenges:

🔗 Medium Post

The module is currently under review (Pull Request #10550) in PyTorch Geometric. More demos and docs will be added soon.

Attack of the Blobs!

Oct. 2025 | CS 248B, Fundamentals of Computer Graphics: Animation and Simulation

Implemented a 2D deformable-body simulation featuring hundreds of interactive “blobs” with realistic elasticity and collision dynamics. Built robust internal forces including stretch, bending, and area-preserving compression, and developed stable collision processing using penalty springs and contact impulses. Designed broad-phase acceleration with bounding volumes to handle dense multi-blob interactions efficiently. Enhanced visual engagement through expressive, deformable blob faces and reactive animations.

The game is available on the left. Press "SPACEBAR" to start.

Enjoy the game!

Kitty Pinball Machine!

Sep. 2025 | CS 248B, Fundamentals of Computer Graphics: Animation and Simulation

Implemented continuous collision detection using SDF ray marching, accurate normal estimation with two-sided finite differences, and interactive flipper mechanics with energy transfer. Focused on robust numerical simulation and engaging gameplay design.

The game is available on the left. Press "SPACEBAR" to start, "F" to flip the left plipper, and "J" to flip the right flipper.

Enjoy the game!