畳み込みニューラル ネットワーク (CNN) の動作を説明するのに役立つ多くの視覚化手法が作成されていますが、それらの大部分は限られた情報を伝える静的図で構成されています。インタラクティブな視覚化は、より豊富な洞察を提供し、ユーザーがモデルの動作をより簡単に調査できるようにします。ただし、それらは通常、簡単に再利用できず、特定のモデルに固有のものです。 Visual Feature Search を紹介します。これは、任意の CNN に一般化でき、研究者のワークフローに簡単に組み込むことができる、新しいインタラクティブな視覚化です。私たちのツールを使用すると、ユーザーは画像領域を強調表示し、特定のデータセットから最も類似した CNN 機能を持つ画像を検索できます。効率的なキャッシュベースの検索実装により、大規模な画像データセットの検索をサポートします。教師あり、自己教師あり、および人間が編集した CNN で実験を実行することにより、モデルの動作のさまざまな側面をツールがどのように解明するかを示します。また、移植可能な Python ライブラリといくつかの IPython ノートブックをリリースして、研究者が独自の実験でツールを簡単に使用できるようにします。私たちのコードは https://github.com/Lookingglasslab/VisualFeatureSearch にあります。
Many visualization techniques have been created to help explain the behavior of convolutional neural networks (CNNs), but they largely consist of static diagrams that convey limited information. Interactive visualizations can provide more rich insights and allow users to more easily explore a model's behavior; however, they are typically not easily reusable and are specific to a particular model. We introduce Visual Feature Search, a novel interactive visualization that is generalizable to any CNN and can easily be incorporated into a researcher's workflow. Our tool allows a user to highlight an image region and search for images from a given dataset with the most similar CNN features. It supports searching through large image datasets with an efficient cache-based search implementation. We demonstrate how our tool elucidates different aspects of model behavior by performing experiments on supervised, self-supervised, and human-edited CNNs. We also release a portable Python library and several IPython notebooks to enable researchers to easily use our tool in their own experiments. Our code can be found at https://github.com/lookingglasslab/VisualFeatureSearch.