410 likes | 499 Views
Modeling Latent Variable Uncertainty for Loss-based Learning. M. Pawan Kumar École Centrale Paris École des Ponts ParisTech INRIA Saclay , Île-de-France. Ben Packer Stanford University. Daphne Koller Stanford University. Aim. Accurate learning with weakly supervised data.
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 • Ongoing and Future Work
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)) User-defined loss ✔ Loss based learning ✖ Loss independent of true (unknown) latent variable ✖ 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,hPθ(y,h|x)
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 log (Pθ(yi|xi))
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
Outline • Previous Methods • Our Framework • Optimization • Results • Ongoing and Future Work
Problem Model Uncertainty in Latent Variables Model Accuracy of Latent Variable Predictions
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 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) (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 • Ongoing and Future Work
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 • Ongoing and Future Work
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 Latent Variable h Running 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
Outline • Previous Methods • Our Framework • Optimization • Results • Ongoing and Future Work