Skip to content

A toolkit that contains advanced tools for Uncertainty Quantification

License

Notifications You must be signed in to change notification settings

aryanator/QuantIQ

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

QuantIQ 🔍

QuantIQ is a unified Python library for building robust, uncertainty-aware deep learning systems. It brings together lightweight, modular tools to help researchers and practitioners gain insight into model reliability and risk through principled uncertainty estimation. poster

📦 What's Inside

Quantiq currently includes two powerful uncertainty quantification tools:

1. DropWise 🔁

A plug-and-play PyTorch/HuggingFace wrapper for Monte Carlo Dropout–based uncertainty estimation in Transformers.

  • Supports classification, regression, QA, and token tagging
  • Computes entropy, confidence, and class-wise variances
  • Enables dropout during inference for Bayesian-style sampling
form quantiq import DropWise

📖 Full DropWise Documentation


2. SmartEnsemble 🧠

A deep ensemble wrapper for PyTorch models with support for adversarial training and dual-mode (epistemic + aleatoric) uncertainty estimation.

  • Works with any PyTorch model
  • Enables risk scoring and calibration
  • Includes built-in visualization and prediction APIs
form quantiq import SmartEnsemble

📖 Full SmartEnsemble Documentation


🔧 Installation

pip install quantiq

Or install from source:

git clone https://github.com/aryanator/QuantIQ.git
cd quantiq
pip install -e .

🧪 Use Cases

  • Safety-critical predictions (medical AI, self-driving, finance)
  • Uncertainty-aware active learning
  • Robust ML pipelines with explainable confidence
  • Research experiments involving confidence, entropy, risk

📚 Documentation & Examples

Explore examples, API usage, and task-specific walkthroughs in the GitHub repository:
🔗 https://github.com/aryanator/QuantIQ

image

📝 License

MIT License


Built by Aryan Patil to make uncertainty estimation simpler, smarter, and production-ready.

About

A toolkit that contains advanced tools for Uncertainty Quantification

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages