Skip to yearly menu bar Skip to main content


Oral

Task-Driven Wavelets using Constrained Empirical Risk Minimization

Eric Marcus · Ray Sheombarsing · Jan-Jakob Sonke · Jonas Teuwen

Summit Ballroom Oral #3
[ ] [ Visit Orals 6A Low-level vision and remote sensing ]
Fri 21 Jun 1:36 p.m. — 1:54 p.m. PDT

Abstract:

Deep Neural Networks (DNNs) are widely used for their ability to effectively approximate large classes of functions. This flexibility, however, makes the strict enforcement of constraints on DNNs a difficult problem. In contexts where it is critical to limit the function space to which certain network components belong, such as wavelets employed in Multi-Resolution Analysis (MRA), naive constraints via additional terms in the loss function are inadequate. To address this, we introduce a Convolutional Neural Network (CNN) wherein the convolutional filters are strictly constrained to be wavelets. This allows the filters to update to task-optimized wavelets during the training procedure. Our primary contribution lies in the rigorous formulation of these filters via a constrained empirical risk minimization framework, thereby providing an exact mechanism to enforce these structural constraints. While our work is grounded in theory, we investigate our approach empirically through applications in medical imaging, particularly in the task of contour prediction around various organs, achieving superior performance compared to baseline methods.

Chat is not available.