Que ferramentas são boas para desenhar diagramas de arquitetura de redes neurais?
Dois programas/serviços recentemente me ajudaram com isso:
- Netron - Leva, por exemplo, um modelo Keras armazenado no formato .h5 e visualiza todas as camadas e parâmetros.
- Upside: Fácil de usar, rápido. Apenas alguns cliques e você tem a sua arquitetura modelada>li>li>Downside: Se você tem uma rede muito profunda, o design torna-se muito "longo", pode não ser muito prático para apresentações ou papers.
- NN SVG: Três estilos: FCNN, LeNet e AlexNet. Para usar no browser, mas também é possível descarregar o SVG depois de ter criado a sua arquitectura. Acabei usando este (estilo AlexNet) para um trabalho de pesquisa recentemente e fiquei muito satisfeito com o visual (fiz uma screenshot para obtê-lo do jeito que eu queria).
- Upside: Permitiu-me definir rapidamente as camadas e ver os resultados ao mesmo tempo. O estilo AlexNet foi muito prático para redes profundas sem ocupar muito espaço.
- Downside: Infelizmente esse estilo não incluiu a opção de nomear as camadas, então eu tive que adicioná-las separadamente depois no Powerpoint. Também com isso você não consegue visualizar outros parâmetros como dropout ou qual função de ativação é aplicada, o que é mostrado quando se usa Netron.