Diff2Dist: Learning Spectrally Distinct Edge Functions, with Applications to Cell Morphology Analysis
This is an incremental method for cell morphology analysis in biology, enabling better graph discrimination for specific datasets.
The authors tackled the problem of discriminating between graphs from wild-type and mutant Arabidopsis thaliana images by learning spectrally descriptive edge weights, achieving improved classification performance with a simple k-nearest-neighbors classifier.
We present a method for learning "spectrally descriptive" edge weights for graphs. We generalize a previously known distance measure on graphs (Graph Diffusion Distance), thereby allowing it to be tuned to minimize an arbitrary loss function. Because all steps involved in calculating this modified GDD are differentiable, we demonstrate that it is possible for a small neural network model to learn edge weights which minimize loss. GDD alone does not effectively discriminate between graphs constructed from shoot apical meristem images of wild-type vs. mutant \emph{Arabidopsis thaliana} specimens. However, training edge weights and kernel parameters with contrastive loss produces a learned distance metric with large margins between these graph categories. We demonstrate this by showing improved performance of a simple k-nearest-neighbors classifier on the learned distance matrix. We also demonstrate a further application of this method to biological image analysis: once trained, we use our model to compute the distance between the biological graphs and a set of graphs output by a cell division simulator. This allows us to identify simulation parameter regimes which are similar to each class of graph in our original dataset.