SpikingJelly笔记之延迟编码

文章目录


前言

记录SpikingJelly中延迟编码的使用方法,以及自定义延时函数。


一、延迟编码的原理

首次脉冲时间编码(Time-to-First-Spike,TTFS)

基于时间的编码方式,输入强度越大,发放时间越早

存在最晚发放时间: t m a x = T t_{max} = T tmax=T

1、线性延迟

python 复制代码
encoder = encoding.LatencyEncoder(T, enc_function='linear')

t f = ( t m a x − 1 ) ( 1 − x ) t_{f} = (t_{max}-1)(1-x) tf=(tmax−1)(1−x)

发放延时 发放时刻

2、对数延迟

python 复制代码
encoder = encoding.LatencyEncoder(T, enc_function='log')

t f = ( t m a x − 1 ) − ln ⁡ ( α x + 1 ) t_{f} = (t_{max}-1)-\ln(\alpha x+1) tf=(tmax−1)−ln(αx+1)

α = exp ⁡ ( t m a x − 1 ) − 1 \alpha = \exp(t_{max}-1)-1 α=exp(tmax−1)−1

t m a x t_{max} tmax不宜过大,防止 α \alpha α溢出

发放延时 发放时刻

二、发放延时转化为脉冲序列

python 复制代码
out_spike = F.one_hot(t_f, T).to(x)

(1)根据输入计算发放时间[0, T-1]

(2)独热编码转为脉冲序列

(3)通过参数设置避免出现全0(不发放)

三、SpikingJelly中的延时编码

python 复制代码
from spikingjelly.activation_based import encoding
encoder = encoding.LatencyEncoder(T=T, enc_function='linear')
# encoder = encoding.LatencyEncoder(T=T, enc_function='log')
# 输出脉冲序列,T:时间步长,w:图像宽度,h:图像高度
out_spike = torch.zeros((T, w, h), dtype=torch.bool)
# 按时间步单步运算
for t in range(T):
	out_spike[t] = encoder(x) # x需要归一化[0,1]
encoder.reset() # 有状态编码器,需要复位

四、自定义延时函数

SpikingJelly中提供了线性和对数两种延迟模式

可以根据需要进行修改,自定义延时函数

在此定义一个指数延时函数: y = − 2.5 + 20.0 ∗ exp ⁡ ( − x 0.5 ) y=-2.5+20.0*\exp(-\frac{x}{0.5}) y=−2.5+20.0∗exp(−0.5x)

当输入过小时允许不发放脉冲

python 复制代码
####################指数延迟####################
class ExpLatencyEncoder(encoding.StatefulEncoder):
    def __init__(self, T: int, step_mode='s'):
        super().__init__(T, step_mode)
    def single_step_encode(self, x: torch.Tensor):
        # 修改发放时间-输入方程
        t_f = -2.5 + 20.0 * torch.exp(-x/0.5)
        # 控制输出范围[0, T]
        t_f = torch.clamp(t_f, 0, T).round().long()
        self.spike = F.one_hot(t_f, num_classes=self.T+1).to(x)
        # [*, T+1] -> [T+1, *]
        d_seq = list(range(self.spike.ndim - 1))
        d_seq.insert(0, self.spike.ndim - 1)
        self.spike = self.spike.permute(d_seq)
        # 截取前T时间步的脉冲
        self.spike = self.spike[:T]
发放延时 发放时刻

五、Lena图像的延迟编码

1、原始图像

读取、展示原始图像

python 复制代码
####################延时编码####################
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt
import torch
import torch.nn.functional as F
from spikingjelly.activation_based import encoding
from spikingjelly import visualizing
####################读取图像####################
img = np.array(Image.open('../dataset/lena.bmp')) / 255
x = torch.from_numpy(img)
w, h = x.shape
plt.figure()
plt.imshow(x, cmap='gray')
plt.axis('off')

2、图像编码

取十个神经元展示编码结果

python 复制代码
####################延迟编码####################
T = 10 # 时间步长
encoder = encoding.LatencyEncoder(T=T, enc_function='linear')
# encoder = encoding.LatencyEncoder(T=T, enc_function='log')
# encoder = ExpLatencyEncoder(T=T)
out_spike = torch.zeros((T, w, h), dtype=torch.bool)
for t in range(T):
    out_spike[t] = encoder(x)
encoder.reset()
# 取十个神经元发放情况
figsize, dpi = (6, 4), 100
visualizing.plot_1d_spikes(spikes=out_spike.numpy()[:,:500:50,250],
                            title='Out Spikes',
                            xlabel='Simulating Step',
                            ylabel='Neuron Index',
                            figsize=figsize,
                            dpi=dpi)
plt.show()
线性延迟 对数延迟 指数延迟(自定义)

对数延时可以将过大和过小的输出区分开,而线性延迟表现得更加均匀一些

3、图像还原

越早发放对应的值越大,因为均采用线性还原,还原质量不代表编码效果

python 复制代码
img_code = torch.zeros(w, h)
for i in range(T):
    img_code += out_spike[i] * (T - i - 1)
plt.imshow(img_code, cmap='gray')
plt.axis('off')
线性延迟 对数延迟 指数延迟(自定义)

总结

延迟编码将输入转化为脉冲序列,是一种基于时间编码的方式

较大的输入对应于较早的输出,有且仅有一次发放

延迟编码的输入需要归一化[0,1]

设置延时函数时需要保证最大延时 t m a x t_{max} tmax小于时间步长 T T T

参考:SpikingJelly:编码器

相关推荐
AI视觉网奇1 小时前
nvlink 训练笔记
pytorch·笔记·深度学习
这个男人是小帅3 小时前
【GCN】 代码详解 (1) 如何运行【pytorch】可运行版本
人工智能·pytorch·python·深度学习·分类
pzx_0013 小时前
【深度学习】梯度累加和直接用大的batchsize有什么区别
pytorch·深度学习
阿_旭4 小时前
实战| 使用深度学习分割和计算水体和农田面积【Pytorch附源码】
人工智能·pytorch·深度学习·ai·目标分割
AI视觉网奇5 小时前
pytorch3d报错:RuntimeError: Not compiled with GPU support.
人工智能·pytorch·python
Rverdoser20 小时前
LSTM:解决梯度消失与长期依赖问题
人工智能·pytorch·python
AI视觉网奇1 天前
显存占用 显存测试
pytorch·python·深度学习
小酒窝.1 天前
深度学习工具 Anaconda、conda、CUDA、cuDNN、pytorch、Cuda Toolkit 解释
pytorch·深度学习·conda
Hoper.J1 天前
PyTorch nn.Embedding() 嵌入详解
pytorch·embedding·嵌入向量·嵌入层
蓝博AI1 天前
基于卷积神经网络的水稻叶片病害识别系统(pytorch框架,python源码)
pytorch·python·cnn