#P16028. [CSPro 23] 脉冲神经网络

[CSPro 23] 脉冲神经网络

题目背景

洛谷的测试数据仅供民间交流使用,非官方测试数据。官方评测链接:https://www.cspro.org/

在本题中,你需要实现一个 SNN(spiking neural network,脉冲神经网络)的模拟器。一个 SNN 由以下几部分组成:

  1. 神经元:按照一定的公式更新内部状态,接受脉冲并可以发放脉冲
  2. 脉冲源:在特定的时间发放脉冲
  3. 突触:连接神经元-神经元或者脉冲源-神经元,负责传递脉冲

题目描述

神经元会按照一定的规则更新自己的内部状态。本题中,我们对时间进行离散化处理,即设置一个时间间隔 Δt\Delta t,仅考虑时间间隔整数倍的时刻 t=kΔt(kZ+)t = k\Delta t (k \in Z^+),按照下面的公式,从 k1k - 1 时刻的取值计算 kk 时刻的变量的取值:

$$\begin{aligned} v_k &= v_{k-1} + \Delta t(0.04v_{k-1}^2 + 5v_{k-1} + 140 - u_{k-1}) + I_k \\ u_k &= u_{k-1} + \Delta t a(bv_{k-1} - u_{k-1}) \end{aligned}$$

其中 vvuu 是神经元内部的变量,会随着时间而变化,aabb 是常量,不会随着时间变化;其中 IkI_k 表示该神经元在 kk 时刻接受到的所有脉冲输入的强度之和,如果没有接受到脉冲,那么 Ik=0I_k = 0。当进行上面的计算后,如果满足 vk30v_k \geq 30,神经元会发放一个脉冲,脉冲经过突触传播到其他神经元;同时,vkv_k 设为 cc 并且 uku_k 设为 uk+du_k + d,其中 ccdd 也是常量。图 1 展示了一个神经元 vv 变量随时间变化的曲线。

:::align{center} :::

突触表示的是神经元-神经元、脉冲源-神经元的连接关系,包含一个入结点和一个出结点(可能出现自环和重边)。当突触的入结点(神经元或者脉冲源)在 kk 时刻发放一个脉冲,那么在传播延迟 D(D>0)D (D > 0) 个时刻以后,也就是在 k+Dk + D 时刻突触的出结点(神经元)会接受到一个强度为 ww 的脉冲。

脉冲源在每个时刻以一定的概率发放一个脉冲,为了模拟这个过程,每个脉冲源有一个参数 0<r32,7670 < r \leq 32,767,并统一采用以下的伪随机函数:

C++ 版本:

static unsigned long next = 1;

/* RAND_MAX assumed to be 32767 */
int myrand(void) {
    next = next * 1103515245 + 12345;
    return((unsigned)(next/65536) % 32768);
}

Python 版本:

next = 1
def myrand():
    global next
    next = (next * 1103515245 + 12345) % (2 ** 64)
    return (next // 65536) % 32768

Java 版本:

long next = 1;
int myrand() {
    next = next * 1103515245 + 12345;
    return (int)((Long.divideUnsigned(next, 65536)) % 32768);
}

在每个时刻,按照编号顺序从小到大,每个脉冲源调用一次上述的伪随机函数,当 r>myrand()r > \text{myrand}() 时,在当前时刻发放一次脉冲,并通过突触传播到神经元。

进行仿真的时候,已知 0 时刻各个神经元的状态,从 1 时刻开始按照上述规则进行计算,直到完成 TT 时刻的计算,再输出 TT 时刻神经元的 vv 值和发放的脉冲次数分别的最小值和最大值。

规定输入数据中结点按如下方式顺序编号:[0,N1][0, N - 1] 为神经元的编号,[N,N+P1][N, N + P - 1] 为脉冲源的编号。

代码中请使用双精度浮点类型。

输入格式

从标准输入读入数据。

输入的第一行包括四个以空格分隔的正整数 N S P TN\ S\ P\ T,表示一共有 NN 个神经元,SS 个突触和 PP 个脉冲源,输出时刻 TT 时神经元的 vv 值。

输入的第二行是一个正实数 Δt\Delta t,表示时间间隔。

输入接下来的若干行,每行有以空格分隔的一个正整数 RNR_N 和六个实数 v u a b c dv\ u\ a\ b\ c\ d,按顺序每一行对应 RNR_N 个具有相同初始状态和常量的神经元:其中 v uv\ u 表示神经元在时刻 0 时的变量取值;a b c da\ b\ c\ d 为该神经元微分方程里的四个常量。保证所有的 RNR_N 加起来等于 NN。它们从前向后按编号顺序描述神经元,每行对应一段连续编号的神经元的信息。

输入接下来的 PP 行,每行是一个正整数 rr,按顺序每一行对应一个脉冲源的 rr 参数。

输入接下来的 SS 行,每行有以空格分隔的两个整数 s(0s<N+P)s (0 \leq s < N + P)t(0t<N)t (0 \leq t < N)、一个实数 w(w0)w (w \geq 0) 和一个正整数 DD,其中 sstt 分别是入结点和出结点的编号;wwDD 分别表示脉冲强度和传播延迟。

输出格式

输出到标准输出。

输出共有两行,第一行由两个近似保留 3 位小数的实数组成,分别是所有神经元在时刻 TT 时变量 vv 的取值的最小值和最大值。第二行由两个整数组成,分别是所有神经元在整个模拟过程中发放脉冲次数的最小值和最大值。

只要按照题目要求正确实现就能通过,不会因为计算精度的问题而得到错误答案。

1 1 1 10
0.1
1 -70.0 -14.0 0.02 0.2 -65.0 2.0
30000
1 0 30.0 2
-35.608 -35.608
2 2
2 4 2 10
0.1
1 -70.0 -14.0 0.02 0.2 -65.0 2.0
1 -69.0 -13.0 0.04 0.1 -60.0 1.0
30000
20000
2 0 15.0 1
3 1 20.0 1
1 0 10.0 2
0 1 40.0 3
-60.000 -22.092
1 2

提示

样例 1 解释

该样例有 1 个神经元、1 个突触和 1 个脉冲源,时间间隔 Δt=0.1\Delta t = 0.1。唯一的脉冲源通过脉冲强度为 30.030.0、传播延迟为 22 的突触传播到唯一的神经元。

该样例一共进行 10 个时间步的模拟,随机数生成器生成 10 次随机数如下:

16838
5758
10113
17515
31051
5627
23010
7419
16212
4086

因此唯一的脉冲源在时刻 1-41\text{-}46-106\text{-}10 发放脉冲。在时间刻从 111010 时,唯一的神经元的 vv 取值分别为:

-70.000
-70.000
-40.000
-8.200
-65.000
-35.404
-32.895
0.181
-65.000
-35.608

该神经元在时刻 55 和时刻 99 发放,最终得到的 v=35.608v = -35.608

子任务

子任务 TT NN SS PP DD 分值
1 102\leq 10^2 30
2 103\leq 10^3 40
3 105\leq 10^5 ^ 10\leq 10 30