1
Uncertainty in Classification
MSCV capstone project
Jia Shi
Advisor: Deva Ramanan
2
Lack of explainability exist in deep learning, decisions process is often unknown
3
Spacecraft
Autonomous Vehicle
Surgical Robots
“One miss shot” will cause catastrophe on safety-critical application
Model can’t say, “I am not sure”. When encounter out-of-distribution(OOD)/noisy sample.
4
Mean: dice
OOD/rare from train set
“In practice, the categorization of uncertainty can be much more complicated than that
and highly task specific”
ID, but noisy data
Mean: knowledge
Model Uncertainty
Data Uncertainty
Two track of project under uncertainty
Data Uncertainty:
Visual Features Disentanglement into Interpretable Attributes Features using Prompts
Question: How do we describe the distribution of data?
(ICCV 2023 submission)
Model Uncertainty:
Improving Generalization with Semantic Consistency Feature
Question: How do we tell whether a model could generalize better on OOD?
(Aim for Neurips 2023 submission)
5
Visual Features Disentanglement into Interpretable Attributes Features using Prompts
Data Uncertainty
6
Question: How do we describe the distribution of data?
human perform classification from visual concept
7
https://arxiv.org/abs/2210.07183, Menon, etc.
Problem of Feature Entanglement
8
end to end training —-> feature entanglement —-> not interpretable
Feature entanglement limit trustworthy
9
Animal
watermelon Skin
Rino body
grass background
…
Visual Feature disentanglement
10
Our goal is to propose a way to disentangle visual feature entanglement to a list of human interpretable attribute
Attribute CLIP - Use prompts to get attributes from original clip embedding
Method: Disentangle Feature
12
Method: Disentangle Feature
13
2. Attribute text encoding with CLIP
Method: Disentangle Feature
14
3. Adopting cosine similarity score as feature!
Visual feature become a weighted combination of interpretable attribute feature
Application: 1. Few shot generalization
15
Using disentangled feature could achieve superior performance on very few shot setting
zero shot evaluation
(time consuming)
(poor performance)
(model ensemble)
Application: 2. Quantifying distribution shift
16
Top attribute from ImageNet-Sketch and ImageNet-Rendition could well reflection the description on those two dataset.
Colorless; Paper; Gray; Light Gray; Lined
ImageNet-Sketch:
17
Application: 2. Quantifying distribution shift
18
Top attribute from ImageNet-Sketch and ImageNet-Rendition could well reflection the description on those two dataset.
Cartoon; Painting; Graffitied; Tattooed;
(ImageNet-R(endition) contains art, cartoons, graffiti, embroidery, graphics, origami, paintings, patterns, plastic objects, plush objects, sculptures, sketches, tattoos, toys, and video game renditions of ImageNet classes.)
ImageNet-Rendition:
Application: 3. Open vocabulary object classification
19
“Swapping” feature weight could adapt to OOD concept without retraining
Our propose dataset:
9 color, 37 fruit, 6050 images
Application: 4. Attribute guided Detection
20
Per-grid classification without accessing the class name.
Potential for Detection Diagnosis
Conclusion
21
Improving Generalization with Semantic Consistency Feature
Model Uncertainty
22
Question: How do we tell whether a model could generalize better on OOD?
How to predict OOD performance?
23
In distribution
OOD distribution
Previous work
24
ID acc -> OOD acc
Miller, etc.
model agreement-> OOD acc
Baek, etc.
https://arxiv.org/abs/2107.04649
https://arxiv.org/abs/2206.13089
Generalization = Semantic Consistency
25
strawberry!
in OOD dataset, for class with same semantic label, there are often shrift in visual appearance
ID vs OOD : Semantic consistency ✅ Visual consistency ❌
example from ImageNet-Rendition
Visual Similarity != Semantic Similarity
26
Supervised method with class-discrimination could not properly preserve semantic consist feature
semantical ideal embedding
(world distribution)
Visual Similarity != Semantic Similarity
27
Supervised method with class-discrimination could not properly preserve semantic consist feature
How to measure the semantic consistency of model embedding?
Semantic Consistency = Mistake Severity
28
Frog
Dog
Cat
Car
Mammal
Animal
Things
What is better mistake?
with target = Cat
Predicting Dog is a better mistake than predicting a Car
Why Semantic Consistency = Mistake Severity
29
GT
Pred 1
Pred 2
Mistake severity is a measurement of how close prediction are semantically close to target
Thus a model with ‘better mistake’ could have more semantically consistent embedding!
Measure (mistake severity using taxonomy loss)
30
+
Current Result
31
Compare to Top 1 ID accuracy (ImageNet acc) , our proposed semantic embedding measurement is a better indicator on performance on OOD dataset ( ImageNet-A acc)
correct trend is monotonically increasing !
Top 1 ID acc =
how separate each class clustering in embedding space
Semantic measurement =
how class cluster embedding are distributed semantically
Conclusion
Future Step
32
Thank you!
33