Support for Annotated Waveforms

[williamsdoug]

I'm hoping to use this thread to discuss how to extend TSAI to support the concept of annotated signals. My motivating example is ECG signal analysis, where one may want to analyze both the entire signal and individual heartbeats within the signal. Signals of this type may contain a set of annotations, where each annotation consists of a position (typically location of QRS complex) and an associated label. In order to perform beat-level analysis it can be useful to extract signal sub-sequences at know offset and width relative the annotated positions within the signal.

I am currently working on the following prototype extensions:

• Static extraction: Preprocessing code to extract subsequences for all or a random subset of annotated points
• Dynamic extraction: A new datatype TSAnnotatedTensor and batch transform TSAnnotationWindow that can extract subsequences at training time.

In addition to extracting signal subsequences, the prototype will also extract label data, supportig the following labeling methods:

• Common label for all annotated instances in the same signal ( row)
• Per-annotation label values
• tensor labels, where return value is subsequence of tensor label using the same offset and width.

While my work is motivated by the ECG analysis use-case, I'd be happy to collaborate with others who have similar use-cases.