具有平坦潜在空间的 PyTorch 几何 GCN 自动编码器

时间:2021-08-01 20:44:25

标签: deep-learning pytorch autoencoder

我有一个问题,其中我有一系列观察结果,每个观察结果都是相同结构的图,但具有不同的节点特征。我想学习每个 32x1 大小的图的平面嵌入。

我的想法是用自动编码器来做到这一点。这将采用输入图,应用一些图卷积,使用密集层将图映射到 32x1 的潜在空间,然后在应用更多卷积之前重建图(使用相同的公共结构)。

据我所知,这与典型的图自动编码器框架形成对比,在典型的图自动编码器框架中,潜在表示是与输入具有相同结构但具有每个节点特征的潜在表示的图。

因此,我不确定如何使用 PyTorch Geometric 实现这样的架构。也就是说,我不确定如何从平坦的潜在空间回到图形。

这可能吗,如果可能,我大概会怎么做?

0 个答案:

没有答案