Hi, I'm Krishna Kumar Bais.

I build uncertainty-aware models

M.Tech in Computer Science & Engineering at IIT Kanpur. I work at the intersection of deep learning, privacy-preserving computation, and scientific visualization — building calibrated, secure, and interpretable systems.

01

About me

I'm a graduate researcher at IIT Kanpur, drawn to problems where reliability, privacy, and interpretability matter — especially in safety-critical vision and data-sensitive systems.

My recent work extends Deformable DETR with aleatoric & epistemic uncertainty, reducing calibration error by 72% while preserving mAP. Alongside, I build privacy-preserving pipelines using secret sharing, MPC, and PIR, and parallel scientific-visualization tools with MPI + VTK.

I've served as Teaching Assistant for 4 courses at IIT Kanpur — from first-year programming to DevOps and databases — and I enjoy mentoring, visualizing data, and shipping end-to-end prototypes.

02

Education

  1. 2024 — Present

    M.Tech, Computer Science & Engineering

    Indian Institute of Technology, Kanpur

    Coursework: Deep Learning for Computer Vision · Machine Learning · Big Data for Visual Analytics · Cryptography Techniques for Privacy Preservation · HCI.

  2. 2018 — 2022

    B.Tech, Computer Science & Engineering

    GRD Institute of Management & Technology, Dehradun

03

Selected projects

Uncertainty-Aware Deformable DETR

CS776 · Prof. Priyanka Bagade · Jan '25 – Apr '25

Extended Deformable DETR with aleatoric variance prediction, MC dropout (T=15) for epistemic uncertainty, and uncertainty-weighted fine-tuning. Reduced Expected Calibration Error by 72% (0.128 → 0.035) on COCO val2017 while preserving mAP (43.3 vs 43.8) and improving APS.

  • PyTorch
  • Deformable DETR
  • MC Dropout
  • COCO

Binary Classification on Combined Datasets

CS771 · Prof. Piyush Rai · Sep '24 – Oct '24

Hybrid pipeline combining text embeddings + LSTM, PCA-reduced deep features, and SVM ensemble with a Random Forest meta-learner.

  • LSTM
  • PCA
  • SVM
  • Random Forest

Detection of Malicious Mobile Webpages

Self Project · Feb '25 – Apr '25

Phishing detector using 21 URL/HTML/JS/traffic features augmented with Naïve-Bayes probability. Random Forest hit 96% accuracy with ~3.6% false positives.

  • scikit-learn
  • Feature Engineering
  • RF · KNN · SVM

Resume Scoring System

Self Project · Jun '25 – Jul '25

End-to-end resume ↔ JD scorer parsing PDFs, extracting entities via NLP, and ranking with semantic similarity. Reached ~0.70 correlation with human ratings.

  • spaCy
  • Sentence Transformers
  • TF-IDF

Continual Learning for Image Classification

CS771 · Prof. Piyush Rai · Oct '24 – Nov '24

Prototype regularization + generative replay on CIFAR-10 with Gaussian prototype classifier; PCA pipeline for incremental updates.

  • TensorFlow
  • scikit-learn
  • PCA

Fairness Analysis for Regression

Self Project · Aug '25

Linear-regression pipeline for absenteeism with group-wise fairness auditing. Bias mitigation achieved 0.00 MAE/RMSE/prediction gap across age, education, and service-time groups.

  • scikit-learn
  • Fairness Metrics
  • Bias Mitigation

Privacy-Preserving Recommendation System

Self Project · Sep '25

Matrix-factorization recommender with two-party additive secret sharing. Uses PIR to securely fetch item features and update secret-shared user profiles — simulating secure multi-party computation end-to-end.

  • MPC
  • Secret Sharing
  • PIR
  • Python

Secure MPC with Beaver Triples

Self Project · May '25 – Jun '25

4-party secret-shared multiplication over TCP sockets using Beaver triples. Implemented Basic, Sequential, and Parallel protocol variants with a triple generator and share distribution.

  • MPC
  • Beaver Triples
  • Sockets

Private Information Retrieval (ASS & SSS)

Self Project · Jul '25 – Aug '25

Privacy-preserving retrieval via PIR with Additive and Shamir's Secret Sharing. Client-server protocols benchmark query creation, reconstruction time, and bandwidth across schemes.

  • PIR
  • Shamir SS
  • Additive SS

Shamir's Secret Sharing

Self Project · May '25

Threshold-based secret splitting and reconstruction using polynomial interpolation — validated recovery from any t shares.

  • Cryptography
  • Polynomial Interpolation

COVID-19 Vaccination Dashboard

CS661 · Prof. Soumya Dutta · Mar '25 – Apr '25

Interactive Streamlit dashboard analyzing global vaccination trends, manufacturer dynamics, and correlations with health outcomes. Maps, scatter, line, bar, and pie views over cleaned, merged datasets.

  • Streamlit
  • Plotly
  • Pandas

Parallel Volume Rendering using MPI

CS677 · Prof. Soumya Dutta, Prof. Preeti Malakar · Oct '24 – Nov '24

Distributed ray-casting with early termination and sub-image compositing. Scalability / timing analysis on 1000×1000×200 and 2000×2000×400 volumes.

  • mpi4py
  • NumPy
  • PIL

Scientific Visualization Pipeline with MPI

CS677 · Sep '24

MPI-parallelized renderer with 1D slicing + 2D block decomposition, colour/opacity transfer functions, and final composited output.

  • MPI
  • Transfer Functions

Isocontour & 3D Volume Rendering

CS661 · Feb '25

Isocontour extraction from raw grids exported to ParaView, plus VTK volume rendering with vtkSmartVolumeMapper, custom transfer functions, and Phong shading.

  • VTK
  • ParaView

Interactive 3D Scientific Data Viewer

CS661 · Mar '25

Jupyter tool linking isosurface rendering with live histogram updates through an isovalue slider, plus reset & reproducible UI controls.

  • VTK
  • NumPy
  • Jupyter

3D Flow Field Path Tracing

CS661 · Apr '25

Particle path tracing in 3D flow fields using RK4 integration — forward and backward streamlines from seed points with vector interpolation.

  • VTK
  • RK4

Uber Priority Ride System

CS798H · Prof. Sruti Srinivasa Ragavan · Mar '25 – Apr '25

HCI design study: identified 45% urgent-ride cancellations as a pain-point and proposed keyword-based detection, priority tags for drivers, and restricted cancellations. Built functional UI prototypes for both customer and driver flows.

  • HCI
  • User Research
  • Prototyping

Gesture & Voice Multimedia System

CS724 · Prof. Amitangshu Pal · Nov '24 – Dec '24

Camera controller for photo / HDR / video via hand gestures and voice commands, plus music playback controls — running on a Raspberry Pi with ultrasonic sensor, Mediapipe, and speech recognition.

  • Raspberry Pi
  • Mediapipe
  • Speech Recognition

Human Posture Detection (PhonePi+)

CS724 · Sep '24

Posture classifier (sitting / standing / lying) from PhonePi+ accelerometer data using threshold-based analysis and visualization.

  • Pandas
  • Matplotlib
  • Sensors

GPS-Based User Localization Simulation

CS724 · Aug '24

Simulated GPS localization from satellite signal travel times; analyzed timing-error impact with visualized error trends.

  • NumPy
  • Matplotlib

Sorting Algorithm Visualizer

Self Project · May '25

Interactive visualizer for Selection, Bubble, Merge, Quick, and Insertion sort — animated states for comparing, swapping, sorted, and pivot, with speed/size sliders and play controls.

  • JavaScript
  • HTML · CSS
  • DOM
04

Skills & toolkit

Languages

C C++ Python SQL

ML / DL Libraries

PyTorch TensorFlow scikit-learn NumPy Pandas Matplotlib Plotly PIL

Visualization & HPC

VTK ParaView Streamlit mpi4py multiprocessing

Tools & Platforms

Git GitHub Jupyter VS Code LaTeX Linux macOS Windows

Relevant Coursework

Deep Learning for Computer Vision Machine Learning Big Data for Visual Analytics Cryptography for Privacy Preservation Human–Computer Interaction Operating Systems DSA OOP Computer Networks DBMS
05

Responsibilities

Teaching Assistant

Software DevOps — CS253

IIT Kanpur · Jan '25 – Apr '25

Designed & evaluated an OOP-in-C++ assignment on memory-efficient file indexing. Reviewed 230+ submissions, assisted exam invigilation and quiz grading.

Teaching Assistant

Embedded & Cyber-Physical Systems — CS637

IIT Kanpur · Aug '24 – Present

Supported 120+ students across grading, project evaluation, and invigilation. Mentored 3 groups on research-paper evaluation projects.

Teaching Assistant

Principles of Database Systems — CS315

IIT Kanpur · Jan '25 – May '25

Grading, malpractice review, and course-administration support.

Teaching Assistant

Fundamentals of Computing — ESC 101

IIT Kanpur · Aug '24 – Dec '24

Mentored first-year undergraduates in a cohort of 600+ students/semester — doubts, grading, and malpractice checks.

Student Guide

New Postgraduate Students

IIT Kanpur · Aug '24 – Present

Part of a 250+ member guide team helping new PG students with academic and social transition at IIT Kanpur.

06

Research paper presentations

Query-driven Visualization of Large Data Sets

CS677 · Sep '24

Explained how query-driven methods prioritize user-defined subsets for targeted visualization. Analyzed the DEX framework's bitmap indexing for fast, scalable exploration of large datasets.

Banner: Image Sensor Reconfiguration Framework

CS724 · Nov '24

How Banner enables seamless resolution switching in image sensors — framework architecture balancing resolution, bandwidth, and energy for mobile/embedded vision.

07

Get in touch

Open to research collaborations, internships, and interesting conversations. The fastest way to reach me is email.