version 0.01
This commit is contained in:
193
src/type.jl
193
src/type.jl
@@ -17,18 +17,19 @@ abstract type knowledgeFn <: Ironpen end
|
||||
#------------------------------------------------------------------------------------------------100
|
||||
|
||||
Base.@kwdef mutable struct kfn_1 <: knowledgeFn
|
||||
params::Dict = Dict() # store params of knowledgeFn itself for later use
|
||||
params::Union{Dict, Nothing} = nothing # store params of knowledgeFn itself for later use
|
||||
|
||||
timeStep::AbstractArray = [0]
|
||||
learningStage::AbstractArray = [0] # 0 inference, 1 start, 2 during, 3 end learning
|
||||
timeStep::Union{AbstractArray, Nothing} = nothing
|
||||
learningStage::Union{AbstractArray, Nothing} = nothing # 0 inference, 1 start, 2 during, 3 end learning
|
||||
zit::Union{AbstractArray, Nothing} = nothing # 3D activation matrix
|
||||
|
||||
# ---------------------------------------------------------------------------- #
|
||||
# LIF Neurons #
|
||||
# ---------------------------------------------------------------------------- #
|
||||
# a projection of kfn.zit into lif dimension for broadcasting later)
|
||||
lif_zit::Union{AbstractArray, Nothing} = nothing
|
||||
lif_zit::Union{AbstractArray, Nothing} = nothing
|
||||
|
||||
# main variables according to papers
|
||||
lif_wRec::Union{AbstractArray, Nothing} = nothing
|
||||
lif_vt0::Union{AbstractArray, Nothing} = nothing
|
||||
lif_vt1::Union{AbstractArray, Nothing} = nothing
|
||||
@@ -39,16 +40,27 @@ Base.@kwdef mutable struct kfn_1 <: knowledgeFn
|
||||
lif_refractoryCounter::Union{AbstractArray, Nothing} = nothing
|
||||
lif_refractoryDuration::Union{AbstractArray, Nothing} = nothing
|
||||
lif_alpha::Union{AbstractArray, Nothing} = nothing
|
||||
lif_delta::AbstractFloat = 1.0
|
||||
lif_tau_m::AbstractFloat = 20.0
|
||||
lif_delta::Union{AbstractFloat, Nothing} = nothing
|
||||
lif_tau_m::Union{AbstractFloat, Nothing} = nothing
|
||||
lif_phi::Union{AbstractArray, Nothing} = nothing
|
||||
lif_epsilonRec::Union{AbstractArray, Nothing} = nothing
|
||||
# lif_eRec::Union{AbstractArray, Nothing} = nothing
|
||||
lif_eRec::Union{AbstractArray, Nothing} = nothing
|
||||
lif_eta::Union{AbstractArray, Nothing} = nothing
|
||||
lif_gammaPd::Union{AbstractArray, Nothing} = nothing
|
||||
lif_wRecChange::Union{AbstractArray, Nothing} = nothing
|
||||
|
||||
lif_firingCounter::Union{AbstractArray, Nothing} = nothing
|
||||
lif_arraySize::Union{AbstractArray, Nothing} = nothing
|
||||
|
||||
# pre-allocation array
|
||||
lif_arrayProjection3DTo4D::Union{AbstractArray, Nothing} = nothing # use to project 3d array to 4d
|
||||
lif_recSignal::Union{AbstractArray, Nothing} = nothing
|
||||
lif_decayed_Vt0::Union{AbstractArray, Nothing} = nothing
|
||||
lif_decayed_EpsilonRec::Union{AbstractArray, Nothing} = nothing
|
||||
lif_vt1_diff_vth::Union{AbstractArray, Nothing} = nothing
|
||||
lif_vt1_diff_vth_div_vth::Union{AbstractArray, Nothing} = nothing
|
||||
lif_gammaPd_div_vth::Union{AbstractArray, Nothing} = nothing
|
||||
lif_phiActivation::Union{AbstractArray, Nothing} = nothing
|
||||
|
||||
# ---------------------------------------------------------------------------- #
|
||||
# ALIF Neurons #
|
||||
@@ -66,22 +78,24 @@ Base.@kwdef mutable struct kfn_1 <: knowledgeFn
|
||||
alif_refractoryCounter::Union{AbstractArray, Nothing} = nothing
|
||||
alif_refractoryDuration::Union{AbstractArray, Nothing} = nothing
|
||||
alif_alpha::Union{AbstractArray, Nothing} = nothing
|
||||
alif_delta::AbstractFloat = 1.0
|
||||
alif_tau_m::AbstractFloat = 20.0
|
||||
alif_delta::Union{AbstractFloat, Nothing} = nothing
|
||||
alif_tau_m::Union{AbstractFloat, Nothing} = nothing
|
||||
alif_phi::Union{AbstractArray, Nothing} = nothing
|
||||
alif_epsilonRec::Union{AbstractArray, Nothing} = nothing
|
||||
alif_epsilonRecA::Union{AbstractArray, Nothing} = nothing
|
||||
# alif_eRec::Union{AbstractArray, Nothing} = nothing
|
||||
alif_eRec::Union{AbstractArray, Nothing} = nothing
|
||||
alif_eta::Union{AbstractArray, Nothing} = nothing
|
||||
alif_gammaPd::Union{AbstractArray, Nothing} = nothing
|
||||
alif_wRecChange::Union{AbstractArray, Nothing} = nothing
|
||||
|
||||
alif_firingCounter::Union{AbstractArray, Nothing} = nothing
|
||||
alif_arraySize::Union{AbstractArray, Nothing} = nothing
|
||||
alif_arrayProjection3DTo4D::Union{AbstractArray, Nothing} = nothing # use to project 3d array to 4d
|
||||
|
||||
alif_a::Union{AbstractArray, Nothing} = nothing # threshold adaptation
|
||||
alif_beta::Union{AbstractArray, Nothing} = nothing # β, constant, value from paper
|
||||
alif_rho::Union{AbstractArray, Nothing} = nothing # ρ, threshold adaptation decay factor
|
||||
alif_tau_a::AbstractFloat = 100.0 # τ_a, adaption time constant in millisecond
|
||||
alif_tau_a::Union{AbstractFloat, Nothing} = nothing # τ_a, adaption time constant in millisecond
|
||||
|
||||
# ---------------------------------------------------------------------------- #
|
||||
# Output Neurons #
|
||||
@@ -99,8 +113,8 @@ Base.@kwdef mutable struct kfn_1 <: knowledgeFn
|
||||
on_refractoryCounter::Union{AbstractArray, Nothing} = nothing
|
||||
on_refractoryDuration::Union{AbstractArray, Nothing} = nothing
|
||||
on_alpha::Union{AbstractArray, Nothing} = nothing
|
||||
on_delta::AbstractFloat = 1.0
|
||||
on_tau_m::AbstractFloat = 20.0
|
||||
on_delta::Union{AbstractFloat, Nothing} = nothing
|
||||
on_tau_m::Union{AbstractFloat, Nothing} = nothing
|
||||
on_phi::Union{AbstractArray, Nothing} = nothing
|
||||
on_epsilonRec::Union{AbstractArray, Nothing} = nothing
|
||||
on_eRec::Union{AbstractArray, Nothing} = nothing
|
||||
@@ -112,13 +126,17 @@ Base.@kwdef mutable struct kfn_1 <: knowledgeFn
|
||||
on_bChange::Union{AbstractArray, Nothing} = nothing
|
||||
|
||||
on_firingCounter::Union{AbstractArray, Nothing} = nothing
|
||||
on_arraySize::Union{AbstractArray, Nothing} = nothing
|
||||
on_arrayProjection3DTo4D::Union{AbstractArray, Nothing} = nothing # use to project 3d array to 4d
|
||||
end
|
||||
|
||||
# outer constructor
|
||||
function kfn_1(params::Dict)
|
||||
function kfn_1(params::Dict; device=cpu)
|
||||
kfn = kfn_1()
|
||||
kfn.params = params
|
||||
|
||||
kfn.timeStep = [0] |> device
|
||||
kfn.learningStage = [0] |> device
|
||||
|
||||
# ---------------------------------------------------------------------------- #
|
||||
# initialize activation matrix #
|
||||
# ---------------------------------------------------------------------------- #
|
||||
@@ -130,7 +148,7 @@ function kfn_1(params::Dict)
|
||||
col += kfn.params[:computeNeuron][:alif][:numbers][2]
|
||||
|
||||
# activation matrix
|
||||
kfn.zit = zeros(row, col, batch)
|
||||
kfn.zit = zeros(row, col, batch) |> device
|
||||
|
||||
# ---------------------------------------------------------------------------- #
|
||||
# LIF config #
|
||||
@@ -138,22 +156,6 @@ function kfn_1(params::Dict)
|
||||
# In 3D LIF matrix, z-axis represent each neuron while each 2D slice represent that neuron's
|
||||
# synaptic subscription to other neurons (via activation matrix)
|
||||
n = kfn.params[:computeNeuron][:lif][:numbers][1] * kfn.params[:computeNeuron][:lif][:numbers][2]
|
||||
kfn.lif_zit = zeros(row, col, n, batch)
|
||||
kfn.lif_vt0 = zeros(1, 1, n, batch)
|
||||
kfn.lif_vt1 = zeros(1, 1, n, batch)
|
||||
kfn.lif_vth = ones(1, 1, n, batch)
|
||||
kfn.lif_vRest = zeros(1, 1, n, batch)
|
||||
kfn.lif_zt0 = zeros(1, 1, n, batch)
|
||||
kfn.lif_zt1 = zeros(1, 1, n, batch)
|
||||
kfn.lif_refractoryCounter = zeros(1, 1, n, batch)
|
||||
kfn.lif_refractoryDuration = ones(1, 1, n, batch) .* 3
|
||||
kfn.lif_alpha = ones(1, 1, n, batch) .* (exp(-kfn.lif_delta / kfn.lif_tau_m))
|
||||
kfn.lif_phi = zeros(1, 1, n, batch)
|
||||
kfn.lif_epsilonRec = zeros(row, col, n, batch)
|
||||
# kfn.lif_eRec = zeros(row, col, n, batch)
|
||||
kfn.lif_eta = zeros(1, 1, n, batch)
|
||||
kfn.lif_gammaPd = zeros(1, 1, n, batch) .* 0.3
|
||||
kfn.lif_wRecChange = zeros(row, col, n, batch)
|
||||
|
||||
# subscription
|
||||
w = zeros(row, col, n)
|
||||
@@ -166,35 +168,67 @@ function kfn_1(params::Dict)
|
||||
end
|
||||
end
|
||||
# project 3D w into 4D kfn.lif_wRec
|
||||
kfn.lif_wRec = reshape(w, (row, col, n, 1)) .* ones(row, col, n, batch)
|
||||
kfn.lif_firingCounter = zeros(1, 1, n, batch)
|
||||
kfn.lif_wRec = reshape(w, (row, col, n, 1)) .* ones(row, col, n, batch) |> device
|
||||
|
||||
kfn.lif_zit = similar(kfn.lif_wRec) .= 0 |> device
|
||||
kfn.lif_vt0 = zeros(1, 1, n, batch) |> device
|
||||
kfn.lif_vt1 = zeros(1, 1, n, batch) |> device
|
||||
kfn.lif_vth = ones(1, 1, n, batch) |> device
|
||||
kfn.lif_vRest = zeros(1, 1, n, batch) |> device
|
||||
# kfn.lif_zt0 = zeros(1, 1, n, batch) |> device
|
||||
kfn.lif_zt1 = zeros(1, 1, n, batch) |> device
|
||||
kfn.lif_refractoryCounter = zeros(1, 1, n, batch) |> device
|
||||
kfn.lif_refractoryDuration = ones(1, 1, n, batch) .* 3 |> device
|
||||
kfn.lif_delta = 1.0
|
||||
kfn.lif_tau_m = 20.0
|
||||
kfn.lif_alpha = ones(1, 1, n, batch) .* (exp(-kfn.lif_delta / kfn.lif_tau_m)) |> device
|
||||
kfn.lif_phi = zeros(1, 1, n, batch) |> device
|
||||
kfn.lif_epsilonRec = zeros(row, col, n, batch) |> device
|
||||
# kfn.lif_eRec = zeros(row, col, n, batch)
|
||||
kfn.lif_eta = zeros(1, 1, n, batch) |> device
|
||||
kfn.lif_gammaPd = zeros(1, 1, n, batch) .* 0.3 |> device
|
||||
kfn.lif_wRecChange = zeros(row, col, n, batch) |> device
|
||||
|
||||
kfn.lif_firingCounter = zeros(1, 1, n, batch) |> device
|
||||
kfn.lif_arraySize = [row, col, n, batch] |> device
|
||||
kfn.lif_arrayProjection3DTo4D = ones(row, col, n, batch) |> device
|
||||
|
||||
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------- #
|
||||
# ALIF config #
|
||||
# ---------------------------------------------------------------------------- #
|
||||
n = kfn.params[:computeNeuron][:alif][:numbers][1] * kfn.params[:computeNeuron][:alif][:numbers][2]
|
||||
kfn.alif_zit = zeros(row, col, n, batch)
|
||||
kfn.alif_vt0 = zeros(1, 1, n, batch)
|
||||
kfn.alif_vt1 = zeros(1, 1, n, batch)
|
||||
kfn.alif_vth = ones(1, 1, n, batch)
|
||||
kfn.alif_avth = ones(1, 1, n, batch)
|
||||
kfn.alif_vRest = zeros(1, 1, n, batch)
|
||||
kfn.alif_zt0 = zeros(1, 1, n, batch)
|
||||
kfn.alif_zt1 = zeros(1, 1, n, batch)
|
||||
kfn.alif_refractoryCounter = zeros(1, 1, n, batch)
|
||||
kfn.alif_refractoryDuration = ones(1, 1, n, batch) .* 3
|
||||
kfn.alif_alpha = ones(1, 1, n, batch) .* (exp(-kfn.alif_delta / kfn.alif_tau_m))
|
||||
kfn.alif_phi = zeros(1, 1, n, batch)
|
||||
kfn.alif_epsilonRec = zeros(row, col, n, batch)
|
||||
kfn.alif_epsilonRecA = zeros(row, col, n, batch)
|
||||
kfn.alif_zit = zeros(row, col, n, batch) |> device
|
||||
kfn.alif_vt0 = zeros(1, 1, n, batch) |> device
|
||||
kfn.alif_vt1 = zeros(1, 1, n, batch) |> device
|
||||
kfn.alif_vth = ones(1, 1, n, batch) |> device
|
||||
kfn.alif_avth = ones(1, 1, n, batch) |> device
|
||||
kfn.alif_vRest = zeros(1, 1, n, batch) |> device
|
||||
# kfn.alif_zt0 = zeros(1, 1, n, batch) |> device
|
||||
kfn.alif_zt1 = zeros(1, 1, n, batch) |> device
|
||||
kfn.alif_refractoryCounter = zeros(1, 1, n, batch) |> device
|
||||
kfn.alif_refractoryDuration = ones(1, 1, n, batch) .* 3 |> device
|
||||
kfn.alif_delta = 1.0
|
||||
kfn.alif_tau_m = 20.0
|
||||
kfn.alif_alpha = ones(1, 1, n, batch) .* (exp(-kfn.alif_delta / kfn.alif_tau_m)) |> device
|
||||
kfn.alif_phi = zeros(1, 1, n, batch) |> device
|
||||
kfn.alif_epsilonRec = zeros(row, col, n, batch) |> device
|
||||
kfn.alif_epsilonRecA = zeros(row, col, n, batch) |> device
|
||||
# kfn.alif_eRec = zeros(row, col, n, batch)
|
||||
kfn.alif_eta = zeros(1, 1, n, batch)
|
||||
kfn.alif_gammaPd = zeros(1, 1, n, batch) .* 0.3
|
||||
kfn.alif_wRecChange = zeros(row, col, n, batch)
|
||||
kfn.alif_eta = zeros(1, 1, n, batch) |> device
|
||||
kfn.alif_gammaPd = zeros(1, 1, n, batch) .* 0.3 |> device
|
||||
kfn.alif_wRecChange = zeros(row, col, n, batch) |> device
|
||||
|
||||
kfn.alif_a = zeros(1, 1, n, batch)
|
||||
kfn.alif_beta = zeros(1, 1, n, batch) .* 0.15
|
||||
kfn.alif_rho = zeros(1, 1, n, batch) .* (exp(-kfn.alif_delta / kfn.alif_tau_a))
|
||||
kfn.alif_a = zeros(1, 1, n, batch) |> device
|
||||
kfn.alif_beta = zeros(1, 1, n, batch) .* 0.15 |> device
|
||||
kfn.alif_tau_a = 100.0
|
||||
kfn.alif_rho = zeros(1, 1, n, batch) .* (exp(-kfn.alif_delta / kfn.alif_tau_a)) |> device
|
||||
|
||||
kfn.alif_firingCounter = zeros(1, 1, n, batch) |> device
|
||||
kfn.alif_arraySize = [row, col, n, batch] |> device
|
||||
kfn.alif_arrayProjection3DTo4D = ones(row, col, n, batch) |> device
|
||||
|
||||
# subscription
|
||||
w = zeros(row, col, n)
|
||||
@@ -207,31 +241,37 @@ function kfn_1(params::Dict)
|
||||
end
|
||||
end
|
||||
# project 3D w into 4D kfn.alif_wRec
|
||||
kfn.alif_wRec = reshape(w, (row, col, n, 1)) .* ones(row, col, n, batch)
|
||||
kfn.alif_firingCounter = zeros(1, 1, n, batch)
|
||||
kfn.alif_wRec = reshape(w, (row, col, n, 1)) .* ones(row, col, n, batch) |> device
|
||||
|
||||
|
||||
# ---------------------------------------------------------------------------- #
|
||||
# output config #
|
||||
# ---------------------------------------------------------------------------- #
|
||||
n = kfn.params[:outputPort][:numbers][1] * kfn.params[:outputPort][:numbers][2]
|
||||
kfn.on_zit = zeros(row, col, n, batch)
|
||||
kfn.on_vt0 = zeros(1, 1, n, batch)
|
||||
kfn.on_vt1 = zeros(1, 1, n, batch)
|
||||
kfn.on_vth = ones(1, 1, n, batch)
|
||||
kfn.on_vRest = zeros(1, 1, n, batch)
|
||||
kfn.on_zt0 = zeros(1, 1, n, batch)
|
||||
kfn.on_zt1 = zeros(1, 1, n, batch)
|
||||
kfn.on_refractoryCounter = zeros(1, 1, n, batch)
|
||||
kfn.on_refractoryDuration = ones(1, 1, n, batch) .* 0
|
||||
kfn.on_alpha = ones(1, 1, n, batch) .* (exp(-kfn.on_delta / kfn.on_tau_m))
|
||||
kfn.on_phi = zeros(1, 1, n, batch)
|
||||
kfn.on_epsilonRec = zeros(row, col, n, batch)
|
||||
kfn.on_zit = zeros(row, col, n, batch) |> device
|
||||
kfn.on_vt0 = zeros(1, 1, n, batch) |> device
|
||||
kfn.on_vt1 = zeros(1, 1, n, batch) |> device
|
||||
kfn.on_vth = ones(1, 1, n, batch) |> device
|
||||
kfn.on_vRest = zeros(1, 1, n, batch) |> device
|
||||
# kfn.on_zt0 = zeros(1, 1, n, batch) |> device
|
||||
kfn.on_zt1 = zeros(1, 1, n, batch) |> device
|
||||
kfn.on_refractoryCounter = zeros(1, 1, n, batch) |> device
|
||||
kfn.on_refractoryDuration = ones(1, 1, n, batch) .* 0 |> device
|
||||
kfn.on_delta = 1.0
|
||||
kfn.on_tau_m = 20.0
|
||||
kfn.on_alpha = ones(1, 1, n, batch) .* (exp(-kfn.on_delta / kfn.on_tau_m)) |> device
|
||||
kfn.on_phi = zeros(1, 1, n, batch) |> device
|
||||
kfn.on_epsilonRec = zeros(row, col, n, batch) |> device
|
||||
# kfn.on_eRec = zeros(row, col, n, batch)
|
||||
kfn.on_eta = zeros(1, 1, n, batch)
|
||||
kfn.on_gammaPd = zeros(1, 1, n, batch) .* 0.3
|
||||
kfn.on_wOutChange = zeros(row, col, n, batch)
|
||||
kfn.on_b = randn(1, 1, n, batch)
|
||||
kfn.on_bChange = randn(1, 1, n, batch)
|
||||
kfn.on_eta = zeros(1, 1, n, batch) |> device
|
||||
kfn.on_gammaPd = zeros(1, 1, n, batch) .* 0.3 |> device
|
||||
kfn.on_wOutChange = zeros(row, col, n, batch) |> device
|
||||
# kfn.on_b = randn(1, 1, n, batch) |> device
|
||||
# kfn.on_bChange = randn(1, 1, n, batch) |> device
|
||||
|
||||
kfn.on_firingCounter = zeros(1, 1, n, batch) |> device
|
||||
kfn.on_arraySize = [row, col, n, batch] |> device
|
||||
kfn.on_arrayProjection3DTo4D = ones(row, col, n, batch) |> device
|
||||
|
||||
# subscription
|
||||
w = zeros(row, col, n)
|
||||
@@ -244,11 +284,8 @@ function kfn_1(params::Dict)
|
||||
end
|
||||
end
|
||||
# project 3D w into 4D kfn.on_wOut
|
||||
kfn.on_wOut = reshape(w, (row, col, n, 1)) .* ones(row, col, n, batch)
|
||||
kfn.on_firingCounter = zeros(1, 1, n, batch)
|
||||
|
||||
|
||||
|
||||
kfn.on_wOut = reshape(w, (row, col, n, 1)) .* ones(row, col, n, batch) |> device
|
||||
|
||||
|
||||
|
||||
return kfn
|
||||
|
||||
Reference in New Issue
Block a user