A mathematical theory for understanding when abstract representations emerge in neural networks
Wang, Johnston, Fusi
Recent experiments reveal that task-relevant variables are often encoded in approximately orthogonal subspaces of the neural activity space. These disentangled low-dimensional representations are observed in multiple brain areas and across different species, and are typically the result of a process of abstraction that supports simple forms of out-of-distribution generalization. The mechanisms by which such geometries emerge remain poorly understood, and the mechanisms that have been investigated are typically unsupervised (e.g., based on variational auto-encoders). Here, we show mathematically that abstract representations of latent variables are guaranteed to appear in the last hidden layer of feedforward nonlinear networks when they are trained on tasks that depend directly on these latent variables. These abstract representations reflect the structure of the desired outputs or the semantics of the input stimuli. To investigate the neural representations that emerge in these networks, we develop an analytical framework that maps the optimization over the network weights into a mean-field problem over the distribution of neural preactivations. Applying this framework to a finite-width ReLU network, we find that its hidden layer exhibits an abstract representation at all global minima of the task objective. We further extend these analyses to two broad families of activation functions and deep feedforward architectures, demonstrating that abstract representations naturally arise in all these scenarios. Together, these results provide an explanation for the widely observed abstract representations in both the brain and artificial neural networks, as well as a mathematically tractable toolkit for understanding the emergence of different kinds of representations in task-optimized, feature-learning network models.
academic
نظرية رياضية لفهم متى تظهر التمثيلات المجردة في الشبكات العصبية
تبحث هذه الورقة عن الآليات الرياضية لظهور التمثيلات المجردة (abstract representations) في الشبكات العصبية. كشفت التجارب أن المتغيرات ذات الصلة بالمهمة عادة ما يتم ترميزها في فضاءات فرعية متعامدة تقريباً من فضاء النشاط العصبي، مما يشكل تمثيلات منخفضة الأبعاد مفكوكة الارتباط. يدعم هذا الهيكل الهندسي التعميم البسيط خارج التوزيع، لكن آلية ظهوره لا تزال غير واضحة. يثبت المؤلفون رياضياً أن التمثيلات المجردة تظهر حتماً في الطبقة المخفية الأخيرة عندما يتم تدريب الشبكات غير الخطية ذات التغذية الأمامية على مهام تعتمد على متغيرات كامنة. لهذا الغرض، طور المؤلفون إطار عمل تحليلي يحول تحسين أوزان الشبكة إلى مشكلة متوسط حقل على توزيعات التنشيط المسبق العصبي.
عمومية التمثيلات المجردة: تشير الدراسات العصبية إلى أن النشاط العصبي في مناطق دماغية متعددة وأنواع مختلفة يظهر تمثيلات مجردة، حيث يتم ترميز المتغيرات ذات الصلة بالمهمة في فضاءات فرعية متعامدة تقريباً
نقص فهم الآلية: على الرغم من انتشار هذا الهيكل الهندسي على نطاق واسع، فإن آليات الشبكة لظهوره لا تزال غير واضحة
قيود الطرق الموجودة: تركزت الآليات المدروسة في الغالب على الطرق غير الموجهة (مثل أجهزة الترميز التلقائي المتغيرة)، لكن بسبب مشاكل التعريف، يصعب على التعلم غير الموجه البحت فك ارتباط التمثيلات