torrvision / focal_calibration
Code for the paper "Calibrating Deep Neural Networks using Focal Loss"
☆155Updated 10 months ago
Related projects ⓘ
Alternatives and complementary repositories for focal_calibration
- Pytorch library for model calibration metrics and visualizations as well as recalibration methods. In progress!☆68Updated 6 months ago
- Calibration of Convolutional Neural Networks☆158Updated last year
- Reliability diagrams visualize whether a classifier model needs calibration☆137Updated 2 years ago
- Code for Deterministic Neural Networks with Appropriate Inductive Biases Capture Epistemic and Aleatoric Uncertainty☆129Updated last year
- Code to accompany the paper 'Improving model calibration with accuracy versus uncertainty optimization'.☆53Updated last year
- Calibration library and code for the paper: Verified Uncertainty Calibration. Ananya Kumar, Percy Liang, Tengyu Ma. NeurIPS 2019 (Spotlig…☆143Updated 2 years ago
- Pitfalls of In-Domain Uncertainty Estimation and Ensembling in Deep Learning, ICLR 2020☆235Updated last year
- Reusable BatchBALD implementation☆74Updated 8 months ago
- Last-layer Laplace approximation code examples☆80Updated 3 years ago
- ☆65Updated 4 years ago
- ☆117Updated 2 years ago
- Gradient Starvation: A Learning Proclivity in Neural Networks☆60Updated 3 years ago
- Code for using CDEP from the paper "Interpretations are useful: penalizing explanations to align neural networks with prior knowledge" ht…☆127Updated 3 years ago
- Code for "On Feature Collapse and Deep Kernel Learning for Single Forward Pass Uncertainty".☆111Updated 2 years ago
- Wrapper for a PyTorch classifier which allows it to output prediction sets. The sets are theoretically guaranteed to contain the true cla…☆229Updated last year
- Code implementation of our ICLR'21 paper "Calibration of Neural Networks using Splines"☆22Updated last year
- Code for "Uncertainty Estimation Using a Single Deep Deterministic Neural Network"☆268Updated 2 years ago
- Combating hidden stratification with GEORGE☆62Updated 3 years ago
- Official PyTorch implementation of the Fishr regularization for out-of-distribution generalization☆83Updated 2 years ago
- Implementation of Estimating Training Data Influence by Tracing Gradient Descent (NeurIPS 2020)☆219Updated 2 years ago
- Robust Out-of-distribution Detection in Neural Networks☆72Updated 2 years ago
- Original dataset release for CIFAR-10H☆82Updated 4 years ago
- ☆21Updated last year
- Implementation of Barlow Twins paper☆99Updated 2 years ago
- A pytorch implementation of our jacobian regularizer to encourage learning representations more robust to input perturbations.☆123Updated last year
- ☆109Updated 2 years ago
- Code for "Supermasks in Superposition"☆117Updated last year
- Addressing Failure Prediction by Learning Model Confidence☆167Updated last year
- Wasserstein Adversarial Active Learning☆29Updated 4 years ago
- Model-agnostic posthoc calibration without distributional assumptions☆42Updated last year