神经网络如何从零学会物理规律：弹跳球轨迹预测实验

一个展示神经网络如何从纯位置数据中学习物理规律（重力、动量、碰撞）的教育项目，无需任何预设物理方程。

• 原作者/维护者：dylaneulgen
• 来源平台：github
• 原始标题：neural-physics-approximator
• 原始链接：https://github.com/dylaneulgen/neural-physics-approximator
• 来源发布时间/更新时间：2026-06-16T01:44:13Z

原作者与来源

• 原作者/维护者：dylaneulgen
• 来源平台：github
• 原始标题：neural-physics-approximator
• 原始链接：https://github.com/dylaneulgen/neural-physics-approximator
• 来源发布时间/更新时间：2026-06-16T01:44:13Z 原作者与来源\n\n- 原作者/维护者： dylaneulgen\n- 来源平台： GitHub\n- 原标题： neural-physics-approximator\n- 原文链接： https://github.com/dylaneulgen/neural-physics-approximator\n- 发布时间： 2026年6月\n\n---\n\n背景：AI 能自己发现物理定律吗？\n\n传统物理模拟依赖于人类总结的数学方程——牛顿定律、能量守恒、碰撞公式。这些方程经过数百年验证，精确而可靠。但一个有趣的问题是：如果神经网络从未接触过这些方程，仅通过观察物体的运动轨迹，能否自己"悟出"背后的物理规律？\n\nneural-physics-approximator 项目正是探索这一问题的实验性实现。它展示了一个小型神经网络如何从纯粹的二维位置数据中学习弹跳球的物理行为，包括重力作用、动量保持和边界碰撞。\n\n---\n\n项目概述：纯数据驱动的物理学习\n\n这个项目的核心设计非常简洁：\n\n1. 一个模拟器生成大量弹跳球的轨迹数据\n2. 神经网络只接收位置坐标 (x, y)，从未接触速度、加速度或物理方程\n3. 网络学习从 10 帧位置预测接下来的 10 帧位置\n4. 为了做出准确预测，网络必须自行推断出重力、动量和碰撞规律\n\n这种设计迫使神经网络成为"物理规律的发现者"而非"方程的执行者"。\n\n---\n\n数据生成：欧拉积分的弹跳球模拟\n\n数据生成模块（csv_generation.py）实现了一个经典的弹跳球模拟器：\n\n- 在一个二维盒子中放置一个球\n- 球受重力影响下落\n- 碰到边界时反弹，每次碰撞损失少量能量\n- 使用欧拉积分计算运动轨迹\n\n模拟器生成数千条轨迹，每条轨迹记录球在不同时刻的 (x, y) 坐标。这些纯位置数据构成了神经网络的训练集。值得注意的是，网络永远不会看到速度向量、重力常量或反弹系数——它必须从位置变化的模式中推断出这一切。\n\n---\n\n网络架构：简洁的三层 MLP\n\n模型采用了一个相对简单的多层感知机（MLP）结构：\n\n\n输入层：20 个神经元（10 帧 × 2 维坐标）\n隐藏层 1：128 个神经元\n隐藏层 2：128 个神经元\n隐藏层 3：128 个神经元\n输出层：20 个神经元（预测接下来 10 帧的坐标）\n\n\n这种架构选择体现了"奥卡姆剃刀"原则——用尽可能简单的模型来测试神经网络的学习能力。如果这样一个基础 MLP 能够从纯位置数据中学到物理规律，那么更复杂的架构应该能做得更好。\n\n---\n\n训练过程：从轨迹到预测\n\n训练流程分为几个关键步骤：\n\n数据预处理\n\ndataset.py 负责加载 CSV 数据，将每条轨迹切分为输入-输出对：输入是连续的 10 帧位置，输出是接下来的 10 帧位置。所有坐标被归一化到 [0, 1] 区间，便于神经网络学习。\n\n训练配置\n\n- 损失函数：均方误差（MSE）\n- 优化器：Adam\n- 数据划分：80% 训练集，20% 测试集（按轨迹划分，确保测试轨迹与训练轨迹完全不同）\n\n学习机制\n\n网络通过最小化预测位置与真实位置的误差来调整权重。为了减小误差，网络必须学会识别以下模式：\n\n- 连续帧之间位置变化的趋势（暗示速度）\n- 位置变化率的变化（暗示加速度/重力）\n- 边界处的突变（暗示碰撞和反弹）\n- 能量逐渐衰减（暗示摩擦/能量损失）\n\n---\n\n可视化：真实 vs 预测的对比\n\n项目提供了一个基于 pygame 的交互式可视化工具（visualize.py），让用户直观感受网络的学习成果：\n\n- 橙色轨迹：真实模拟器的物理计算结果\n- 蓝色轨迹：神经网络的预测结果\n\n用户可以在窗口中点击任意位置投放一个新球，观察两个轨迹的对比。初始阶段，两条轨迹几乎重合；随着时间推移，预测误差累积，蓝色轨迹会逐渐偏离橙色轨迹。\n\n这种"自回归预测"模式特别有趣：网络用自己的预测结果作为下一步的输入，持续向前推演。这意味着任何微小的误差都会被放大，最终预测轨迹与真实物理轨迹分道扬镳。\n\n---\n\n关键洞察：神经网络学到了什么？\n\n这个实验揭示了几个关于神经网络学习物理规律的深刻见解：\n\n1. 隐式物理表示\n\n网络没有显式存储"重力 = 9.8 m/s²"这样的公式，但它在权重中编码了对物理行为的隐式理解。输入位置序列时，它能输出符合物理直觉的预测。\n\n2. 误差累积现象\n\n自回归预测中的误差累积是一个普遍问题。即使单次预测误差很小，多次迭代后预测轨迹也会显著偏离真实轨迹。这与混沌系统的敏感性相呼应。\n\n3. 数据质量的重要性\n\n网络的表现完全取决于训练数据的质量和多样性。如果训练数据没有涵盖某些边界情况（比如特定角度的碰撞），网络在遇到这些情况时就会失效。\n\n---\n\n应用场景与延伸思考\n\n虽然这是一个教育性质的实验项目，但其思路有广泛的应用潜力：\n\n物理引擎加速\n\n传统物理模拟需要逐帧求解方程，计算成本较高。训练好的神经网络可以作为"近似物理引擎"，用前向传播替代方程求解，在精度要求不高的场景下大幅提升性能。\n\n复杂系统建模\n\n对于难以用简单方程描述的复杂系统（如流体、软体、人群），数据驱动的方法可能比手工建模更有效。\n\nAI 科学发现\n\n这个项目是"AI 辅助科学发现"的一个缩影。未来，类似的系统或许能从实验数据中自动发现新的物理定律，辅助人类科学家进行研究。\n\n---\n\n项目结构与技术细节\n\n项目代码组织清晰，每个模块职责单一：\n\n- config.py：物理常量和训练超参数\n- csv_generation.py：轨迹数据生成器\n- dataset.py：数据加载和预处理\n- model.py：神经网络定义\n- train.py：训练脚本\n- visualize.py：交互式可视化\n- requirements.txt：依赖列表\n- weights.pth：预训练权重（可直接运行可视化）\n\n项目依赖简单，仅需标准的 Python 科学计算库（PyTorch、NumPy、Pygame）。预训练权重的包含让用户可以立即体验可视化效果，无需等待训练完成。\n\n---\n\n总结：从数据到理解的桥梁\n\nneural-physics-approximator 是一个优雅的教育项目，它用最简单的设置展示了神经网络的强大能力：从纯观察数据中学习复杂的因果规律。\n\n这个项目提醒我们，神经网络的"黑盒"特性既是挑战也是机遇。虽然我们无法直接解读网络学到的"物理定律"，但它确实能够做出符合物理直觉的预测。这种从数据到行为的映射，正是现代机器学习的核心思想。\n\n对于学习深度学习的开发者而言，这是一个极佳的入门项目。它代码简洁、概念清晰，同时又触及了物理模拟、序列预测、自回归生成等多个重要主题。