Ritabrata Chakraborty

Ritabrata Chakraborty

ML Intern

Uber

About Me

I’m Ritabrata Chakraborty, a final-year student at BITS Pilani pursuing Mechanical Engineering with a Minor in Data Science.

My journey began with game development in high school and has since evolved into a deep focus on machine learning, robotics, and computer vision. Currently, I work at Uber as a Machine Learning Intern, developing foundation model-driven annotation pipelines that support autonomous vehicle perception systems.

Beyond my current role, I’ve worked on projects involving UAV navigation, multi-robot coordination, synthetic data generation, and uncertainty-guided path planning. My goal is to bring cutting-edge ML research into practical robotics applications, bridging the gap between academic innovation and real-world deployment.


Open Collaboration & Opportunities

I’m currently seeking funding and collaborators for two ongoing initiatives:

  • 🏑 Automated Hockey Ball Launcher - Planned hardware-software system for controlled shot generation and data collection. Progress limited by hardware and funding. Seeking collaborators to advance this project.
    Proposal

  • 🌑 3D Lunar/Mars Terrain Modelling for Rover Simulation - Proposed DEM-based 3D planetary terrain environment for rover navigation research. Open to collaboration from planetary robotics experts.
    Proposal

Beyond these specific projects, I’m open to internships, research partnerships, and other professional opportunities in robotics, computer vision, and autonomous systems. Whether you’re interested in collaboration, have project ideas, or want to discuss potential roles, I’d be glad to connect.


Download my Résumé.

Interests
  • Machine Learning & Deep Learning
  • Computer Vision & Robotics
  • Autonomous Systems & UAV Navigation
  • Multi-Robot Coordination
  • Path Planning & SLAM
  • Generative AI (Predictive Maintenance)
Education
  • B.E. in Mechanical Engineering — Minor in Data Science, 2022–2026

    Birla Institute of Technology and Science, Pilani (BITS Pilani)

Experience

 
 
 
 
 
Uber
ML Intern
Jul 2025 – Dec 2025 Bangalore, India

Supervisor: Mr. Siddarth Malreddy, Tech Lead Manager & Mr. Ishan Nigam, Senior ML Engineer, Uber

  • Augmented uLabel with deep learning object tracking for automated RGB/IR video annotation, reducing manual effort
  • Deployed XGBoost anomaly detection in human-in-the-loop tracking validation for frame-level annotation accuracy

Tech Stack: PyTorch, Python, C++

 
 
 
 
 
National University of Singapore (NUS)
Research Intern
Mar 2025 – Present Singapore (Remote)

Supervisor: Dr. Guillaume Sartoretti, Assistant Professor, MARMoT Lab, NUS

Vision-Attention-Driven Autonomous Navigation with Semantic Understanding

  • Enhanced CogniPlan with cross-attention between frontier and node embeddings to enrich node representations
  • Incorporated Visual Navigation Transformer (ViNT) to capture semantic context for adaptive exploration strategies

Uncertainty-Guided Path Planning via Conditional Layout Prediction

  • Architected CogniPlan, integrating Wasserstein GAN-based conditional inpainting model and graph attention network for uncertainty-aware navigation
  • Achieved up to 17.7% shorter exploration paths and 3.9% improved navigation efficiency over state-of-the-art baselines across 100+ unseen maps, using a lightweight model with fewer than 0.35M parameters enabling real-time CPU inference

Tech Stack: ROS Noetic (with Gazebo, Rviz), PyTorch, Python

 
 
 
 
 
KU Leuven
Research Intern
Sep 2024 – Present Belgium (Remote)

Supervisor: Dr. Pradeep Kundu, Assistant Professor, KU Leuven

Auxiliary Classifier WGAN-GP for Time-Series Sensor Data Generation

  • Built ACWGAN-GP with TCN discriminator to augment minority fault classes, reaching ~100% classification accuracy
  • Evaluated synthetic data by comparing real and generated FFT spectra using PCC, Cosine Similarity, MMD, and KL Div

Conditional Latent Diffusion-GAN for CWT Generation and LiteFormer-based Classification

  • Developed Conditional Latent Denoising Diffusion GAN (LDDGAN) with Supervised Contrastive Loss (SCL) for latent class separation (99.9% AE accuracy, 16× compression with EvoNorm-S0)
  • Designed LiteFormer variants integrating Continuous Wavelet Convolution (CWC), Haar DWT, WDCNN-BiLSTM-Siamese Network, and CWMS-GAN-inspired architectures, achieving up to 99.18% fault classification accuracy

Tech Stack: PyTorch, Python

 
 
 
 
 
CSIR-CEERI
Research Intern
Jan 2024 – Present Rajasthan, India

Supervisor: Dr. Kaushal Kishore, Principal Scientist, CSIR-CEERI

Exploring Unknown Indoor Environments with Drones Using a Multi-Critic DDPG Architecture

  • Orchestrated a multi-critic DDPG control system for continuous pitch–yaw–roll drone navigation in indoor spaces
  • Implemented a hybrid reward structure with artificial potential fields and precision replay into DDPG, improving navigation

Monocular Vision-Based UAV Navigation for Orchard Monitoring

  • Engineered a UAV-based orchard monitoring system using YOLOv11 (Box mAP50: 95.5%, Mask mAP50: 96.5%)
  • Programmed B-spline trajectory logic and implemented custom yaw-roll controller to minimize drift under wind

Tech Stack: PyTorch, Python, ROS Noetic, AirSim, NVIDIA Jetson Orin NX

Projects

Some of the things I have built in the past!

*
Multi-Critic DDPG for Autonomous Indoor Drone Navigation

Multi-Critic DDPG for Autonomous Indoor Drone Navigation

A novel approach combining importance sampling with DDPG for improved deep reinforcement learning performance and sample efficiency.

3D Lunar Surface Modeling Using Chandrayaan-2 TMC DEM Data

3D Lunar Surface Modeling Using Chandrayaan-2 TMC DEM Data

Advanced planetary simulation system leveraging Unreal Engine 5 and Chandrayaan-2 Terrain Mapping Camera data for high-fidelity lunar surface modeling and rover mission planning applications.

Vision-Attention-Driven Autonomous Navigation with Semantic Understanding

Vision-Attention-Driven Autonomous Navigation with Semantic Understanding

CogniPlan enhancement with cross-attention mechanisms and ViNT integration for semantic-aware adaptive exploration.

An Efficient Approach for Synthetic Data Generation and Fault Diagnosis for Rotating Machinery

An Efficient Approach for Synthetic Data Generation and Fault Diagnosis for Rotating Machinery

This study explores the use of an Auxiliary Classifier Wasserstein GAN with Gradient Penalty (ACWGAN-GP) for synthetic data generation and fault diagnosis in rotating machinery, addressing data scarcity and class imbalance challenges.

Large-Scale DRL Exploration with Tunable Adaptive Exploration-Exploitation

Large-Scale DRL Exploration with Tunable Adaptive Exploration-Exploitation

Deep reinforcement learning-based reactive planner for large-scale Lidar-based autonomous robot exploration in 2D action space with novel alpha conditioning for exploration-exploitation control.

Conditional Latent Diffusion-GAN for Synthetic CWT Generation and LiteFormer-based Fault Classification in Rotating Machinery

Conditional Latent Diffusion-GAN for Synthetic CWT Generation and LiteFormer-based Fault Classification in Rotating Machinery

Advanced GAN architecture combining Conditional Latent Diffusion Denoising with Continuous Wavelet Transform generation for enhanced fault classification in industrial systems.

MathWorks Global Student Drone Challenge 2025

MathWorks Global Student Drone Challenge 2025

3rd place autonomous drone navigation system using vision-based control with masking, ray-tracing, and closed-loop yaw control.

Path Planning of Low-Altitude UAV for Tree Canopy Tracking and Orchard Monitoring

Path Planning of Low-Altitude UAV for Tree Canopy Tracking and Orchard Monitoring

Monocular vision-based UAV treeline following system using instance segmentation and curve-fitting algorithms for precise orchard navigation under wind disturbances.

CogniPlan: Uncertainty-Guided Path Planning with Conditional Generative Layout Prediction

CogniPlan: Uncertainty-Guided Path Planning with Conditional Generative Layout Prediction

CogniPlan leverages conditional generative inpainting models to predict multiple plausible layouts, mirroring human cognitive maps for uncertainty-guided path planning in unknown environments.

Computer Vision Pipeline for 3D Trajectory Reconstruction and Shot Classification

Computer Vision Pipeline for 3D Trajectory Reconstruction and Shot Classification

3D ball trajectory reconstruction achieving 0.76m RMSE and shot classification with 96.4% accuracy using TCN-attention.

Physics-Based Launcher Mechanical Design and FEM Analysis

Physics-Based Launcher Mechanical Design and FEM Analysis

FEM-validated hockey ball launcher achieving 150 km/h with CFRP composite design and safety factor 2.07 at 496 RPM.

European Rover Challenge (ERC) Remote 2023

European Rover Challenge (ERC) Remote 2023

5th place + Best Maintenance Award - ExoMy rover navigation with UR3 arm manipulation using ArUco detection and MoveIt motion planning.

Achievements

Technical Skills

My Technical Skills & Core Competencies

Relevant Courses

• Machine Learning • Deep Learning • Foundations of Data Science • Applied Statistical Methods • Linear Algebra • Computer Programming • Engineering Optimization • Differential Equations • Design of Machine Elements • Digital Twins

Programming Languages

• Python • C++ • C • Shell (Linux)

Robotics & Simulation

• ROS (with Gazebo, Rviz) • MAVROS • Navigation Stack • MoveIt • AirSim • Unity • Unreal Engine • MATLAB • Simulink • QGIS

Machine Learning

• PyTorch • TensorFlow • Scikit-Learn • OpenCV • Open3D • Matplotlib • Weights & Biases (W&B)

Hardware & Embedded Systems

• NVIDIA Jetson (Nano, Orin) • Raspberry Pi • Arduino • IMUs • Stereo Camera • 3D LiDAR

CAD & Mechanical Simulation

• ANSYS Mechanical • SolidWorks • Fusion 360

Recent Posts

Coming Soon!

Hi

A simple greeting post.

Recent Publications

Quickly discover relevant content by filtering publications.
(2026). Multi-Critic DDPG for Autonomous Indoor Drone Navigation. Under Review.

Paper

(2025). Path Planning of Low-Altitude UAV for Tree Canopy Tracking and Orchard Monitoring. Patent (In Progress).

PDF

(2025). CogniPlan: Uncertainty-Guided Path Planning with Conditional Generative Layout Prediction. CoRL 2025.

Project Paper Code

Teaching

Teaching Assistantships at BITS Pilani

 
 
 
 
 
BITS Pilani
Teaching Assistant
Sep 2024 – Present Rajasthan, India

CS F320: Foundations of Data Science
ME F218: Advanced Mechanics of Solids
ME F216: Materials Science and Engineering

  • Assisted 100+ students in labs and tutorials, clarifying concepts and linking theory to practical applications
  • Evaluated assignments and supported faculty in delivering high-impact teaching sessions

Leadership & Community

Leadership Roles & Community Building Activities at BITS Pilani

 
 
 
 
 
Mechanical Engineering Association (MEA), BITS Pilani
President & Secretary
Jun 2024 – Present Rajasthan, India
  • Coordinated 10+ events and career sessions for 300+ students, facilitating technical exposure and alumni interaction.
  • Managed production of 500+ merchandise items and led outreach, boosting student participation by 20%.
 
 
 
 
 
Indian Society of Heating, Refrigerating, and Air Conditioning Engineers (ISHRAE), BITS Pilani
President & Tech Fest Coordinator
Oct 2024 – Jul 2025 Rajasthan, India
  • Led a team of 25+ members to organize 4+ technical workshops with HVAC industry experts.
  • Hosted 3 competitions and networking events, engaging 200+ students in HVAC innovation and awareness.
 
 
 
 
 
Tinkerer's Lab (TL), BITS Pilani
Project Manager
May 2024 – Jul 2025 Rajasthan, India
  • Supervised 5 interdisciplinary robotics teams (30+ members) on Micromouse and Hexapod projects.
  • Oversaw lab resources, conducted weekly reviews, and mentored 50+ students in hands-on technical skills.

Contact