48 lines
2 KiB
C++
48 lines
2 KiB
C++
/**
|
|
* @file spike_generators.cpp
|
|
* @brief Functions for creating specific spikes generators.
|
|
* @kaspersky_support D. Postnikov
|
|
* @date 05.02.2026
|
|
* @license Apache 2.0
|
|
* @copyright © 2026 AO Kaspersky Lab
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
// cppcheck-suppress missingInclude
|
|
#include <cstdlib>
|
|
|
|
#include "global_config.h"
|
|
#include "network_functions.h"
|
|
|
|
|
|
// Generate spike patterns from training labels for AltAI neuron model.
|
|
// This template specialization creates a spike generator that converts discrete label information into spike patterns
|
|
// at specific time intervals during training. The generator sends spikes on the 11th step of each 15-step image
|
|
// processing cycle to avoid conflicts with image data presentation and ensure proper label synchronization.
|
|
template <>
|
|
std::function<knp::core::messaging::SpikeData(knp::core::Step)>
|
|
make_training_labels_spikes_generator<knp::neuron_traits::AltAILIF>(const Dataset& dataset)
|
|
{
|
|
return [&dataset](knp::core::Step step)
|
|
{
|
|
knp::core::messaging::SpikeData message;
|
|
|
|
knp::core::Step local_step = step % steps_per_image;
|
|
// KNP_DELTA (error-driven training): the label is sent on every active step, in step with the image.
|
|
static const bool every_step = std::getenv("KNP_DELTA") != nullptr;
|
|
if (every_step ? local_step < active_steps : local_step == 11)
|
|
message.push_back(dataset.get_data_for_training().first[step / steps_per_image].first);
|
|
return message;
|
|
};
|
|
}
|