Download
1 / 36
360 likes | 376 Views
This paper presents a framework to model uncertainty in latent variables and accuracy of predictions. It addresses the challenge of learning with weakly supervised data by minimizing dissimilarity between distributions. The approach incorporates user-defined loss functions and improves prediction accuracy. The proposed methodology is evaluated through experiments on object detection tasks, showcasing significant results in loss metrics.
E N D
Modeling Latent Variable Uncertainty for Loss-based Learning M. Pawan Kumar ÉcoleCentrale Paris Écoledes PontsParisTech INRIA Saclay, Île-de-France Ben Packer Stanford University Daphne Koller Stanford University
Aim Accurate learning with weakly supervised data Train Input xi Output yi Input x Bison Deer Elephant Giraffe Output y = “Deer” Latent Variable h Llama Object Detection Rhino
Aim Accurate learning with weakly supervised data Input x Feature Ψ(x,y,h) (e.g. HOG) Function f : Ψ(x,y,h) (-∞, +∞) Output y = “Deer” Latent Variable h • (y(f),h(f)) = argmaxy,h f(Ψ(x,y,h)) Prediction
Aim Accurate learning with weakly supervised data Input x Feature Ψ(x,y,h) (e.g. HOG) Function f : Ψ(x,y,h) (-∞, +∞) Output y = “Deer” Latent Variable h • f* = argminfObjective(f) Learning
Aim Find a suitable objective function to learn f* Input x Feature Ψ(x,y,h) (e.g. HOG) Function f : Ψ(x,y,h) (-∞, +∞) Encourages accurate prediction Output y = “Deer” User-specified criterion for accuracy Latent Variable h • f* = argminfObjective(f) Learning
Outline • Previous Methods • Our Framework • Optimization • Results
Latent SVM Linear function parameterized by w Prediction (y(w), h(w)) = argmaxy,hwTΨ(x,y,h) Learning minwΣiΔ(yi,yi(w),hi(w)) ✔ Loss based learning ✖ Loss function has a restricted form ✖ Doesn’t model uncertainty in latent variables
Expectation Maximization exp(θTΨ(x,y,h)) Joint probability Pθ(y,h|x) = Z Prediction (y(θ), h(θ)) = argmaxy,hθTΨ(x,y,h) Learning maxθΣi Σhilog (Pθ(yi,hi|xi)) ✔ Models uncertainty in latent variables ✖ Doesn’t model accuracy of latent variable prediction ✖ No user-defined loss function
Problem Model Uncertainty in Latent Variables Model Accuracy of Latent Variable Predictions
Outline • Previous Methods • Our Framework • Optimization • Results
Solution Use two different distributions for the two different tasks Model Uncertainty in Latent Variables Model Accuracy of Latent Variable Predictions
Solution Use two different distributions for the two different tasks Pθ(hi|yi,xi) hi Model Accuracy of Latent Variable Predictions
Solution Use two different distributions for the two different tasks Pθ(hi|yi,xi) hi Pw(yi,hi|xi) (yi,hi) (yi(w),hi(w))
The Ideal Case No latent variable uncertainty, correct prediction Pθ(hi|yi,xi) hi hi(w) Pw(yi,hi|xi) (yi,hi(w)) (yi,hi)
In Practice Restrictions in the representation power of models Pθ(hi|yi,xi) hi Pw(yi,hi|xi) (yi,hi) (yi(w),hi(w))
Our Framework Minimize the dissimilarity between the two distributions Pθ(hi|yi,xi) hi User-defined dissimilarity measure Pw(yi,hi|xi) (yi,hi) (yi(w),hi(w))
Our Framework Minimize Rao’s Dissimilarity Coefficient Pθ(hi|yi,xi) hi ΣhΔ(yi,h,yi(w),hi(w))Pθ(h|yi,xi) Pw(yi,hi|xi) (yi,hi) (yi(w),hi(w))
Our Framework Minimize Rao’s Dissimilarity Coefficient Pθ(hi|yi,xi) hi Hi(w,θ) - β Σh,h’Δ(yi,h,yi,h’)Pθ(h|yi,xi)Pθ(h’|yi,xi) Pw(yi,hi|xi) (yi,hi) (yi(w),hi(w))
Our Framework Minimize Rao’s Dissimilarity Coefficient Pθ(hi|yi,xi) hi Hi(w,θ) - β Hi(θ,θ) - (1-β) Δ(yi(w),hi(w),yi(w),hi(w)) Pw(yi,hi|xi) (yi,hi) (yi(w),hi(w))
Our Framework Minimize Rao’s Dissimilarity Coefficient Pθ(hi|yi,xi) hi minw,θ Σi Hi(w,θ) - β Hi(θ,θ) Pw(yi,hi|xi) (yi,hi) (yi(w),hi(w))
Outline • Previous Methods • Our Framework • Optimization • Results
Optimization minw,θΣiHi(w,θ) - β Hi(θ,θ) Initialize the parameters to w0 and θ0 Repeat until convergence Fix w and optimize θ Fix θ and optimize w End
Optimization of θ minθΣiΣhΔ(yi,h,yi(w),hi(w))Pθ(h|yi,xi) - β Hi(θ,θ) Pθ(hi|yi,xi) hi hi(w) Case I: yi(w) = yi
Optimization of θ minθΣiΣhΔ(yi,h,yi(w),hi(w))Pθ(h|yi,xi) - β Hi(θ,θ) Pθ(hi|yi,xi) hi hi(w) Case I: yi(w) = yi
Optimization of θ minθΣiΣhΔ(yi,h,yi(w),hi(w))Pθ(h|yi,xi) - β Hi(θ,θ) Pθ(hi|yi,xi) hi Case II: yi(w) ≠ yi
Optimization of θ minθΣiΣhΔ(yi,h,yi(w),hi(w))Pθ(h|yi,xi) - β Hi(θ,θ) Stochastic subgradient descent Pθ(hi|yi,xi) hi Case II: yi(w) ≠ yi
Optimization of w minwΣiΣhΔ(yi,h,yi(w),hi(w))Pθ(h|yi,xi) Expected loss, models uncertainty Form of optimization similar to Latent SVM Concave-Convex Procedure (CCCP) Observation: When Δ is independent of true h, our framework is equivalent to Latent SVM
Outline • Previous Methods • Our Framework • Optimization • Results
Object Detection Train Input xi Output yi Input x Bison Deer Elephant Output y = “Deer” Latent Variable h Giraffe Mammals Dataset Llama 60/40 Train/Test Split 5 Folds Rhino
Results – 0/1 Loss Statistically Significant
Action Detection Train Input xi Output yi Input x Jumping Phoning Playing Instrument Reading Riding Bike Output y = “Using Computer” Riding Horse Running Latent Variable h PASCAL VOC 2011 Taking Photo 60/40 Train/Test Split UsingComputer 5 Folds Walking
Results – 0/1 Loss Statistically Significant
Results – Overlap Loss Statistically Significant
Conclusions • Two separate distributions • Conditional probability of latent variables • Delta distribution for prediction • Generalizes latent SVM • Future work • Large-scale efficient optimization • Distribution over w • New applications
Code available at http://cvc.centrale-ponts.fr/personnel/pawan
