arXiv:2409.02327stat.MLcs.LG2024-09

让低方差结果信息进入潜空间,提升神经调控靶点选择效果

Generative Principal Component Regression via Variational Inference

  • 基于监督变分自编码器设计新目标,强制将关键信息纳入潜变量
  • 在模拟与真实神经数据中,预测性能显著优于传统方法
  • 适合脑科学干预研究,尤其关注低方差行为特征的建模

操控复杂系统(如大脑)以调节特定结果具有深远意义,尤其在精神疾病治疗中。一种设计有效干预策略的方法是针对预测模型的关键特征。尽管生成式潜变量模型(如概率主成分分析,PPCA)是识别干预靶点的强大工具,但其难以将与低方差结果相关的信息融入潜空间。若在该情形下基于潜空间设计刺激靶点,干预效果可能较差。为此,我们提出一种基于监督变分自编码器(SVAE)的新目标,强制将此类信息保留在潜空间中。该目标可与线性模型(如PPCA)结合,形成生成式主成分回归(gPCR)。仿真结果显示,gPCR在干预靶点选择上显著优于标准PCR和SVAE。我们还开发了一个指标,用于检测相关信息是否未被正确纳入载荷矩阵。在两个与压力和社交行为相关的神经数据集中,gPCR的预测性能显著超越PCR,而SVAE在载荷中对相关信息的整合程度较低。总体表明,该方法在使用潜变量模型进行干预靶点选择时,明显优于现有推断方案。

原文摘要 · Abstract (English)

The ability to manipulate complex systems, such as the brain, to modify specific outcomes has far-reaching implications, particularly in the treatment of psychiatric disorders. One approach to designing appropriate manipulations is to target key features of predictive models. While generative latent variable models, such as probabilistic principal component analysis (PPCA), is a powerful tool for identifying targets, they struggle incorporating information relevant to low-variance outcomes into the latent space. When stimulation targets are designed on the latent space in such a scenario, the intervention can be suboptimal with minimal efficacy. To address this problem, we develop a novel objective based on supervised variational autoencoders (SVAEs) that enforces such information is represented in the latent space. The novel objective can be used with linear models, such as PPCA, which we refer to as generative principal component regression (gPCR). We show in simulations that gPCR dramatically improves target selection in manipulation as compared to standard PCR and SVAEs. As part of these simulations, we develop a metric for detecting when relevant information is not properly incorporated into the loadings. We then show in two neural datasets related to stress and social behavior in which gPCR dramatically outperforms PCR in predictive performance and that SVAEs exhibit low incorporation of relevant information into the loadings. Overall, this work suggests that our method significantly improves target selection for manipulation using latent variable models over competitor inference schemes.

生成模型潜变量神经调控因果推断

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。