Imageomics / Imageomics/FuncaPalooza-2025

Augmenting a SOTA object detector using latent spaces from foundation models.

Open
#14 0 comments 0 reactions 0 assignees View on GitHub
dataset project idea
Dominant language
No language data
Stars
5
Forks
0
PR merge metrics
No merged PRs in 30d

Description

# Multi-Modal Multi-Task Continuous Triplet Loss (CTL) for Enhanced Object Detection

## Overview
This project idea explores **augmenting the features from a State-of-the-Art (SOTA) object detector** with a **shared latent space** learned using the **Continuous Triplet Loss (CTL)** method.
As part of my dissertation work, I've developed a CTL method which acts as a **multi-task multi-modal foundation model** by producing a latent representation that captures task-relevant similarities across diverse label types (categorical, continuous, or mixed). The hope is that integrating this latent space into an object detection network (e.g., **YOLO** or **DeepForest**) provides additional features that can improve classification, localization, and regression performance.

## Motivation
Conventional object detectors (YOLO, Faster R-CNN, DeepForest) rely solely on features extracted from image backbones. While effective, these features may not fully capture:
- **Cross-task dependencies** (e.g., classification and regression)
- **Multi-modal cues** (e.g., RGB, hyperspectral, LiDAR)
- **Continuous label relationships** (e.g., tree height, crown width)

The **CTL latent space** addresses these gaps by:
- Aligning multi-task features according to **functional similarity** rather than discrete label matches.
- Preserving **semantic distances** between samples even when labels are continuous or partially related.
- Providing a **modality-aware shared space** that can be injected into downstream networks.

## Method
1. **Train CTL Latent Space (this is already done as part of my dissertation)**
- Input: Multi-modal data (RGB, hyperspectral, LiDAR).
- Tasks: Tree species classification, height regression, bounding box regression, and other forest attributes.
- Loss: Continuous Triplet Loss with task-weighted label similarity.
- Output: Unified latent representation with preserved task-relevant geometry.

2. **Integrate with SOTA Object Detector**
- Choose a detector such as:
- **YOLOv8** for general object detection.
- **DeepForest** for tree detection and species classification.
- Extract detector backbone features and **concatenate or fuse** them with the CTL latent vectors.

3. **Joint Fine-Tuning**
- Fine-tune the combined model on object detection datasets.
- Optimize detector losses and potentially an auxiliary CTL alignment loss.

## Expected Benefits
- **Improved detection accuracy** for visually ambiguous species.
- **Better generalization** to new geographic regions or sensor modalities.
- **Multi-task consistency**, ensuring classification and regression tasks benefit from shared structure.
- **Transferable latent space** that can support related detection tasks without retraining from scratch.

## Potential Applications
- Multi-modal object detection.

## References
- [YOLOv8](https://github.com/ultralytics/ultralytics)
- [DeepForest](https://deepforest.readthedocs.io/en/v1.5.0/)

Contributor guide

No contributing guide indexed for this repository

Research direction

No repository files, tests, or implementation entry point are identified. Start by selecting YOLOv8 or DeepForest and defining how the existing CTL latent vectors will be fused with detector backbone features, then identify suitable detection data and evaluation criteria. Done means a fine-tuned detector can be compared against the original detector for detection and related task performance.

Written by the indexing model from the issue text.

Assessment

Tech stack
machine-learning
Domain
computer-vision, machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
20/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.