You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
* Merge from Main (#56)
* Merge JOSS paper review branch (#54)
* add paper for submission to joss
* Fix missing DOIs
* include code in the examples as `code` in the paper
* Fix link to the license (#46) (#47)
* update reference to tensorflow
* Reduce repetition and clarify newly added features in VisualTorch
* revise the paper v2
* update by pre-commit
* Update LICENSE (#55)
* Add a co-author
* Update paper based on feedback from Paul Gavrikov
* Update paper.md
* Update paper.md
* Update paper.md
* Update paper.md
* Update paper.md
* Update paper.md
* updating affiliations (#59)
* fix precommit
* Add orcid id
---------
Co-authored-by: Paul Gavrikov <paulgavrikov@users.noreply.github.com>
Copy file name to clipboardExpand all lines: paper/paper.md
+15-6Lines changed: 15 additions & 6 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -9,10 +9,19 @@ tags:
9
9
10
10
authors:
11
11
- name: Willy Fitra Hendria
12
-
affiliation: 1
12
+
affiliation: "1"
13
+
orcid: 0000-0002-6209-3981
14
+
- name: Paul Gavrikov
15
+
affiliation: "2, 3"
16
+
orcid: 0000-0003-2667-9333
13
17
affiliations:
14
-
- name: Independent Researcher
15
-
index: 1
18
+
- index: 1
19
+
name: Independent Researcher
20
+
- index: 2
21
+
name: IMLA, Offenburg University, Germany
22
+
- index: 3
23
+
name: University of Mannheim, Germany
24
+
16
25
date: 18 February 2024
17
26
bibliography: paper.bib
18
27
---
@@ -25,13 +34,13 @@ VisualTorch is a library designed for visualizing neural network architectures i
25
34
26
35
Neural network architecture visualization plays an important role in the scientific process within the realm of artificial intelligence and machine learning. While mathematical equations and descriptive paragraphs provide detailed information about architectures, effective visualizations can significantly aid scientists in communicating their proposed architectures to others.
27
36
28
-
In deep learning projects based on Keras [@Chollet:2015], the visualkeras project [@Gavrikov:2020] has been gaining traction over nearly four years of development. It offers visualization of Keras-based neural network architectures in two styles: layered and graph. The visualtorch library provides visualization for PyTorch-based architectures [@Paszke:2019]. PyTorch itself has gained popularity among deep learning frameworks within the scientific community in recent years [@Aoun:2022]. Our VisualTorch library offers more visualization styles, improved usability, and a development environment by providing web-based documentation and CI/CD pipelines for seamless future contributions and collaborations.
37
+
In deep learning projects based on Keras [@Chollet:2015], the visualkeras project [@Gavrikov:2020] has been gaining traction over nearly four years of development. It offers visualization of Keras-based neural network architectures in two styles: layered and graph. The visualtorch library provides visualization for PyTorch-based architectures [@Paszke:2019]. PyTorch itself has gained popularity among deep learning frameworks within the scientific community in recent years [@Aoun:2022]. VisualTorch expands the functionality of VisualKeras to PyTorch and offers more visualization styles, improved usability, and a development environment by providing web-based documentation and CI/CD pipelines for seamless future contributions and collaborations.
29
38
30
39
# Introduction
31
40
32
-
Recent advancements in artificial intelligence have sparked widespread interest among researchers, particularly in exploring innovative algorithmic approaches such as neural networks or deep learning architectures. These architectures have demonstrated remarkable utility across various AI applications, including computer vision, natural language processing, and robotics. For implementing the neural network architectures, many researchers and practitioners often utilize established deep learning frameworks, such as PyTorch [@Paszke:2019], TensorFlow [@Abadi:2016], and Keras [@Chollet:2015]
41
+
Recent advancements in artificial intelligence have sparked widespread interest among researchers, particularly in exploring innovative algorithmic approaches such as neural networks or deep learning architectures. These architectures have demonstrated remarkable utility across various AI applications, including computer vision, natural language processing, and robotics. To implement neural network architectures, many researchers and practitioners often utilize established deep learning frameworks, such as PyTorch [@Paszke:2019], TensorFlow [@Abadi:2016], and Keras [@Chollet:2015]
33
42
34
-
To effectively communicate their ideas, researchers often employ architecture diagrams as aids for comprehension. While detailed mathematical descriptions help in understanding the intricacies of algorithms, visual representations of architectures offer an additional means of conveying concepts, enabling individuals to grasp the overall visual representation. Our VisualTorch library is designed to facilitate the visualization of PyTorch-based neural network architectures. Instead of manually constructing diagrams from scratch, researchers can simply leverage our library to generate visualizations. With a variety of customization options, users can tailor visualizations to suit their preferences.
43
+
To effectively communicate their ideas, practitioners often employ architecture diagrams as aids for comprehension. While detailed mathematical descriptions help in understanding the intricacies of algorithms, visual representations of architectures offer an additional means of conveying concepts, enabling individuals to grasp the overall visual representation. VisualTorch is designed to facilitate the visualization of PyTorch-based neural network architectures. Instead of manually constructing diagrams from scratch, practitioners can simply leverage our library to generate visualizations. With a variety of customization options, users can tailor visualizations to suit their preferences.
35
44
36
45
One of the important features of VisualTorch is its ability to automatically map a neural network model to visualizations using various styles such as layered, graph, and LeNet-like visualization [@Lecun:1998]. Users can further refine these visualizations by adjusting attributes such as color, opacity, and size. VisualTorch aims to offer a solution for rapidly visualizing a wide range of neural network architectures in PyTorch. Inspired by the visualkeras [@Gavrikov:2020] project, our VisualTorch library shares a similar motivation: to assist in visualizing neural network architectures. Unlike visualkeras, our library offers enhanced functionality specifically tailored for PyTorch-based architectures, supporting models defined with `torch.nn.Module` and `torch.nn.Sequential`. In addition to providing more visualization styles, including the recent addition of LeNet style, our library also provides online web-based documentation and streamlined CI/CD workflows, which improve usability and facilitate future development.
0 commit comments