English
Related papers

Related papers: A survey on Kornia: an Open Source Differentiable …

200 papers

We present PyMoosh, a Python-based simulation library designed to provide a comprehensive set of numerical tools allowing to compute essentially all optical characteristics of multilayered structures, ranging from reflectance and…

Visual understanding goes well beyond object recognition. With one glance at an image, we can effortlessly imagine the world beyond the pixels: for instance, we can infer people's actions, goals, and mental states. While this task is easy…

Computer Vision and Pattern Recognition · Computer Science 2019-03-27 Rowan Zellers , Yonatan Bisk , Ali Farhadi , Yejin Choi

This paper presents the design, implementation, and evaluation of the PyTorch distributed data parallel module. PyTorch is a widely-adopted scientific computing package used in deep learning research and applications. Recent advances in…

Distributed, Parallel, and Cluster Computing · Computer Science 2020-06-30 Shen Li , Yanli Zhao , Rohan Varma , Omkar Salpekar , Pieter Noordhuis , Teng Li , Adam Paszke , Jeff Smith , Brian Vaughan , Pritam Damania , Soumith Chintala

Modern imaging instruments can produce terabytes to petabytes of data for a single experiment. The biggest barrier to processing big image datasets has been computational, where image analysis algorithms often lack the efficiency needed to…

Computer Vision and Pattern Recognition · Computer Science 2026-03-13 Nicholas Schaub , Andriy Kharchenko , Hamdah Abbasi , Sameeul Samee , Hythem Sidky , Nathan Hotaling

Automatic differentiation (AD) frameworks such as JAX and PyTorch have enabled gradient-based optimization for a wide range of scientific fields. Yet, many "hard" primitives in these libraries such as thresholding, Boolean logic, discrete…

Machine Learning · Computer Science 2026-03-11 Anselm Paulus , A. René Geist , Vít Musil , Sebastian Hoffmann , Onur Beker , Georg Martius

Multiple Instance Learning (MIL) is a powerful framework for weakly supervised learning, particularly useful when fine-grained annotations are unavailable. Despite growing interest in deep MIL methods, the field lacks standardized tools for…

Recent advancements in code completion models have primarily focused on local file contexts. However, these studies do not fully capture the complexity of real-world software development, which often requires the use of rapidly-evolving…

We present OpenDCVCs, an open-source PyTorch implementation designed to advance reproducible research in learned video compression. OpenDCVCs provides unified and training-ready implementations of four representative Deep Contextual Video…

Image and Video Processing · Electrical Eng. & Systems 2025-08-07 Yichi Zhang , Fengqing Zhu

With the rapid development of VR technology, the demand for high-quality 3D models is increasing. Traditional methods struggle with efficiency and quality in large-scale customization. This paper introduces a deep-learning framework that…

Computer Vision and Pattern Recognition · Computer Science 2024-09-05 Jie Fu , Shun Fu , Mick Grierson

Recent progress in scientific visualization has expanded the scope of visualization from being merely a way of presentation to an analysis and discovery tool. A given visualization result is usually generated by applying a series of…

Machine Learning · Computer Science 2021-08-13 Drishti Maharjan , Peter Zaspel

Deep learning has enabled major advances in the fields of computer vision, natural language processing, and multimedia among many others. Developing a deep learning system is arduous and complex, as it involves constructing neural network…

Machine Learning · Computer Science 2017-08-04 Hao Dong , Akara Supratak , Luo Mai , Fangde Liu , Axel Oehmichen , Simiao Yu , Yike Guo

We explore different approaches to integrating a simple convolutional neural network (CNN) with the Lucene search engine in a multi-stage ranking architecture. Our models are trained using the PyTorch deep learning toolkit, which is…

Information Retrieval · Computer Science 2017-07-27 Zhucheng Tu , Matt Crane , Royal Sequiera , Junchen Zhang , Jimmy Lin

Remotely sensed geospatial data are critical for applications including precision agriculture, urban planning, disaster monitoring and response, and climate change research, among others. Deep learning methods are particularly promising for…

Computer Vision and Pattern Recognition · Computer Science 2022-09-20 Adam J. Stewart , Caleb Robinson , Isaac A. Corley , Anthony Ortiz , Juan M. Lavista Ferres , Arindam Banerjee

The $\texttt{torch-choice}$ is an open-source library for flexible, fast choice modeling with Python and PyTorch. $\texttt{torch-choice}$ provides a $\texttt{ChoiceDataset}$ data structure to manage databases flexibly and…

Machine Learning · Computer Science 2025-06-05 Tianyu Du , Ayush Kanodia , Susan Athey

Current training methods for deep neural networks boil down to very high dimensional and non-convex optimization problems which are usually solved by a wide range of stochastic gradient descent methods. While these approaches tend to work…

Machine Learning · Computer Science 2019-09-30 Avraam Chatzimichailidis , Franz-Josef Pfreundt , Nicolas R. Gauger , Janis Keuper

This paper introduces HyperNOs, a PyTorch library designed to streamline and automate the process of exploring neural operators, with a special focus on hyperparameter optimization for comprehensive and exhaustive exploration. Indeed,…

Machine Learning · Computer Science 2026-02-09 Massimiliano Ghiotto

The non-local module is designed for capturing long-range spatio-temporal dependencies in images and videos. Although having shown excellent performance, it lacks the mechanism to model the interactions between positions across channels,…

Computer Vision and Pattern Recognition · Computer Science 2018-11-02 Kaiyu Yue , Ming Sun , Yuchen Yuan , Feng Zhou , Errui Ding , Fuxin Xu

Large language models are built on top of a transformer-based architecture to process textual inputs. For example, the LLaMA stands out among many open-source implementations. Can the same transformer be used to process 2D images? In this…

Computer Vision and Pattern Recognition · Computer Science 2024-07-09 Xiangxiang Chu , Jianlin Su , Bo Zhang , Chunhua Shen

In this paper, we introduce si4onnx, a package for performing selective inference on deep learning models. Techniques such as CAM in XAI and reconstruction-based anomaly detection using VAE can be interpreted as methods for identifying…

Machine Learning · Computer Science 2025-01-30 Teruyuki Katsuoka , Tomohiro Shiraishi , Daiki Miwa , Shuichi Nishino , Ichiro Takeuchi

Visibility is a fundamental concept in computational geometry, with numerous applications in surveillance, robotics, and games. This software paper presents T\v{r}iVis, a C++ library developed by the authors for computing numerous…

Robotics · Computer Science 2025-02-18 Jan Mikula , Miroslav Kulich , Libor Přeučil