Timezone: »

 
Poster
Visualizing the PHATE of Neural Networks
Scott Gigante · Adam S Charles · Smita Krishnaswamy · Gal Mishne

Tue Dec 10 10:45 AM -- 12:45 PM (PST) @ East Exhibition Hall B + C #173

Understanding why and how certain neural networks outperform others is key to guiding future development of network architectures and optimization methods. To this end, we introduce a novel visualization algorithm that reveals the internal geometry of such networks: Multislice PHATE (M-PHATE), the first method designed explicitly to visualize how a neural network's hidden representations of data evolve throughout the course of training. We demonstrate that our visualization provides intuitive, detailed summaries of the learning dynamics beyond simple global measures (i.e., validation loss and accuracy), without the need to access validation data. Furthermore, M-PHATE better captures both the dynamics and community structure of the hidden units as compared to visualization based on standard dimensionality reduction methods (e.g., ISOMAP, t-SNE). We demonstrate M-PHATE with two vignettes: continual learning and generalization. In the former, the M-PHATE visualizations display the mechanism of "catastrophic forgetting" which is a major challenge for learning in task-switching contexts. In the latter, our visualizations reveal how increased heterogeneity among hidden units correlates with improved generalization performance. An implementation of M-PHATE, along with scripts to reproduce the figures in this paper, is available at https://github.com/scottgigante/M-PHATE.

Author Information

Scott Gigante (Yale University)
Adam S Charles (Princeton University)
Smita Krishnaswamy (Yale University)
Gal Mishne (UC San Diego)

More from the Same Authors

  • 2021 : A sandbox for prediction and integration of DNA, RNA, and proteins in single cells »
    Malte Luecken · Daniel Burkhardt · Robrecht Cannoodt · Christopher Lance · Aditi Agrawal · Hananeh Aliee · Ann Chen · Louise Deconinck · Angela Detweiler · Alejandro Granados · Shelly Huynh · Laura Isacco · Yang Kim · Dominik Klein · BONY DE KUMAR · Sunil Kuppasani · Heiko Lickert · Aaron McGeever · Honey Mekonen · Joaquin Melgarejo · Maurizio Morri · Michaela Müller · Norma Neff · Sheryl Paul · Bastian Rieck · Kaylie Schneider · Scott Steelman · Michael Sterr · Daniel Treacy · Alexander Tong · Alexandra-Chloe Villani · Guilin Wang · Jia Yan · Ce Zhang · Angela Pisco · Smita Krishnaswamy · Fabian Theis · Jonathan M Bloom
  • 2021 Spotlight: Learning Disentangled Behavior Embeddings »
    Changhao Shi · Sivan Schwartz · Shahar Levy · Shay Achvat · Maisan Abboud · Amir Ghanayim · Jackie Schiller · Gal Mishne
  • 2022 : Topological Jet Tagging »
    Dawson Thomas · Sarah Demers · Smita Krishnaswamy · Bastian Rieck
  • 2022 : Geometric NeuralPDE (GNPnet) Models for Learning Dynamics »
    Oluwadamilola Fasina · Smita Krishnaswamy · Aditi Krishnapriyan
  • 2023 Workshop: New Frontiers in Graph Learning (GLFrontiers) »
    Jiaxuan You · Rex Ying · Hanjun Dai · Ge Liu · Azalia Mirhoseini · Smita Krishnaswamy
  • 2022 Spotlight: Lightning Talks 6A-4 »
    Xiu-Shen Wei · Konstantina Dritsa · Guillaume Huguet · ABHRA CHAUDHURI · Zhenbin Wang · Kevin Qinghong Lin · Yutong Chen · Jianan Zhou · Yongsen Mao · Junwei Liang · Jinpeng Wang · Mao Ye · Yiming Zhang · Aikaterini Thoma · H.-Y. Xu · Daniel Sumner Magruder · Enwei Zhang · Jianing Zhu · Ronglai Zuo · Massimiliano Mancini · Hanxiao Jiang · Jun Zhang · Fangyun Wei · Faen Zhang · Ioannis Pavlopoulos · Zeynep Akata · Xiatian Zhu · Jingfeng ZHANG · Alexander Tong · Mattia Soldan · Chunhua Shen · Yuxin Peng · Liuhan Peng · Michael Wray · Tongliang Liu · Anjan Dutta · Yu Wu · Oluwadamilola Fasina · Panos Louridas · Angel Chang · Manik Kuchroo · Manolis Savva · Shujie LIU · Wei Zhou · Rui Yan · Gang Niu · Liang Tian · Bo Han · Eric Z. XU · Guy Wolf · Yingying Zhu · Brian Mak · Difei Gao · Masashi Sugiyama · Smita Krishnaswamy · Rong-Cheng Tu · Wenzhe Zhao · Weijie Kong · Chengfei Cai · WANG HongFa · Dima Damen · Bernard Ghanem · Wei Liu · Mike Zheng Shou
  • 2022 Spotlight: Manifold Interpolating Optimal-Transport Flows for Trajectory Inference »
    Guillaume Huguet · Daniel Sumner Magruder · Alexander Tong · Oluwadamilola Fasina · Manik Kuchroo · Guy Wolf · Smita Krishnaswamy
  • 2022 Competition: Multimodal Single-Cell Integration Across Time, Individuals, and Batches »
    Daniel Burkhardt · Jonathan Bloom · Robrecht Cannoodt · Malte Luecken · Smita Krishnaswamy · Christopher Lance · Angela Pisco · Fabian Theis
  • 2022 : Multimodal Single-Cell Integration Across Time and Individuals »
    Daniel Burkhardt · Smita Krishnaswamy · Robrecht Cannoodt · Malte Luecken · Jonathan Bloom · Fabian Theis · Christopher Lance · Angela Pisco
  • 2022 Poster: Diffusion Curvature for Estimating Local Curvature in High Dimensional Data »
    Dhananjay Bhaskar · Kincaid MacDonald · Oluwadamilola Fasina · Dawson Thomas · Bastian Rieck · Ian Adelstein · Smita Krishnaswamy
  • 2022 Poster: Manifold Interpolating Optimal-Transport Flows for Trajectory Inference »
    Guillaume Huguet · Daniel Sumner Magruder · Alexander Tong · Oluwadamilola Fasina · Manik Kuchroo · Guy Wolf · Smita Krishnaswamy
  • 2022 Poster: DiSC: Differential Spectral Clustering of Features »
    Ram Dyuthi Sristi · Gal Mishne · Ariel Jaffe
  • 2021 : Multimodal Single-Cell Data Integration + Q&A »
    Daniel Burkhardt · Smita Krishnaswamy · Malte Luecken · Debora Marks · Angela Pisco · Bastian Rieck · Jian Tang · Alexander Tong · Fabian Theis · Guy Wolf
  • 2021 Poster: Learning Disentangled Behavior Embeddings »
    Changhao Shi · Sivan Schwartz · Shahar Levy · Shay Achvat · Maisan Abboud · Amir Ghanayim · Jackie Schiller · Gal Mishne
  • 2020 : Closing Remarks »
    Frederic Chazal · Smita Krishnaswamy · Roland Kwitt · Karthikeyan Natesan Ramamurthy · Bastian Rieck · Yuhei Umeda · Guy Wolf
  • 2020 Workshop: Learning Meaningful Representations of Life (LMRL.org) »
    Elizabeth Wood · Debora Marks · Ray Jones · Adji Bousso Dieng · Alan Aspuru-Guzik · Anshul Kundaje · Barbara Engelhardt · Chang Liu · Edward Boyden · Kresten Lindorff-Larsen · Mor Nitzan · Smita Krishnaswamy · Wouter Boomsma · Yixin Wang · David Van Valen · Orr Ashenberg
  • 2020 Workshop: Topological Data Analysis and Beyond »
    Bastian Rieck · Frederic Chazal · Smita Krishnaswamy · Roland Kwitt · Karthikeyan Natesan Ramamurthy · Yuhei Umeda · Guy Wolf
  • 2020 : Opening Remarks »
    Frederic Chazal · Smita Krishnaswamy · Roland Kwitt · Karthikeyan Natesan Ramamurthy · Bastian Rieck · Yuhei Umeda · Guy Wolf
  • 2020 Poster: Uncovering the Topology of Time-Varying fMRI Data using Cubical Persistence »
    Bastian Rieck · Tristan Yates · Christian Bock · Karsten Borgwardt · Guy Wolf · Nicholas Turk-Browne · Smita Krishnaswamy
  • 2020 Spotlight: Uncovering the Topology of Time-Varying fMRI Data using Cubical Persistence »
    Bastian Rieck · Tristan Yates · Christian Bock · Karsten Borgwardt · Guy Wolf · Nicholas Turk-Browne · Smita Krishnaswamy
  • 2019 : Phenotype »
    Nir HaCohen · David Reshef · Matthew Johnson · Sam Morris · Aurel Nagy · Gokcen Eraslan · Meromit Singer · Eliezer Van Allen · Smita Krishnaswamy · Casey Greene · Scott Linderman · Alexander Wiltschko · Dylan Kotliar · James Zou · Brendan Bulik-Sullivan
  • 2018 Poster: Geometry Based Data Generation »
    Ofir Lindenbaum · Jay Stanley · Guy Wolf · Smita Krishnaswamy
  • 2018 Spotlight: Geometry Based Data Generation »
    Ofir Lindenbaum · Jay Stanley · Guy Wolf · Smita Krishnaswamy