Differentiable Spline Approximations

Minsu Cho, Aditya Balu, Ameya Joshi, Anjana Deva Prasad, Biswajit Khara, Soumik Sarkar, Baskar Ganapathysubramanian, Adarsh Krishnamurthy, Chinmay Hegde

    Research output: Chapter in Book/Report/Conference proceedingConference contribution

    Abstract

    The paradigm of differentiable programming has significantly enhanced the scope of machine learning via the judicious use of gradient-based optimization. However, standard differentiable programming methods (such as autodiff) typically require the machine learning models to be differentiable, limiting their applicability. Our goal in this paper is to use a new, principled approach to extend gradient-based optimization to functions well modeled by splines, which encompass a large family of piecewise polynomial models. We derive the form of the (weak) Jacobian of such functions and show that it exhibits a block-sparse structure that can be computed implicitly and efficiently. Overall, we show that leveraging this redesigned Jacobian in the form of a differentiable "layer" in predictive models leads to improved performance in diverse applications such as image segmentation, 3D point cloud reconstruction, and finite element analysis.

    Original languageEnglish (US)
    Title of host publicationAdvances in Neural Information Processing Systems 34 - 35th Conference on Neural Information Processing Systems, NeurIPS 2021
    EditorsMarc'Aurelio Ranzato, Alina Beygelzimer, Yann Dauphin, Percy S. Liang, Jenn Wortman Vaughan
    PublisherNeural information processing systems foundation
    Pages20270-20282
    Number of pages13
    ISBN (Electronic)9781713845393
    StatePublished - 2021
    Event35th Conference on Neural Information Processing Systems, NeurIPS 2021 - Virtual, Online
    Duration: Dec 6 2021Dec 14 2021

    Publication series

    NameAdvances in Neural Information Processing Systems
    Volume24
    ISSN (Print)1049-5258

    Conference

    Conference35th Conference on Neural Information Processing Systems, NeurIPS 2021
    CityVirtual, Online
    Period12/6/2112/14/21

    ASJC Scopus subject areas

    • Computer Networks and Communications
    • Information Systems
    • Signal Processing

    Fingerprint

    Dive into the research topics of 'Differentiable Spline Approximations'. Together they form a unique fingerprint.

    Cite this