Skip to content

feat(cli): enhancements to the auto-benchmark search algorithms - #327

Merged
Syspretor merged 5 commits into
sgl-project:mainfrom
diw-zw:0506-tpe
May 8, 2026
Merged

Syspretor merged 5 commits into
sgl-project:mainfrom
diw-zw:0506-tpe

Conversation

@diw-zw

@diw-zw diw-zw commented May 8, 2026 •

Copy link
Copy Markdown
Collaborator

Overview

This PR introduces significant enhancements to the auto-benchmark search algorithms, focusing on SLA constraint evaluation improvements, and search space handling refinements.

Key Changes

1. Search Algorithm Improvements

  • Removed param.go: Eliminated the old parameter expansion logic that was causing issues in TPE random phase
  • Enhanced Grid Search:
    • Added proper parameter type validation and expansion
    • Support for categorical, int, and float parameters with step values
    • Rejected unsupported types (continuous params without step, pow2)
    • Improved Cartesian product computation with overflow protection

2. TPE/Optuna Integration

  • Optuna Bridge Enhancement:
    • Moved optuna_bridge.py to proper location (tools/optuna/)
    • Improved distribution descriptor handling for different parameter types
    • Better support for float, int, and categorical distributions
  • Search Space Handling:
    • Proper constraint and parameter type processing
    • Raw search space preservation for pow2 decoding

3. SLA Constraint Evaluation

  • Relative Deviation Calculation:
    • Changed from binary pass/fail to relative deviation ratios
    • Max constraints: deviation = max(0, (actual - limit) / limit)
    • Allows Optuna sampler to perceive violation severity
  • Removed EarlyStopPatience:
    • Prevents premature stopping during TPE random exploration phase
  • Percentage-based Metrics:
    • Changed SLA evaluation to use percentage values instead of absolute values

4. Configuration & Parsing

  • Enhanced Config Parsing:
    • Improved search space configuration handling
    • Better validation for parameter definitions
  • Updated Config Types:
    • Added support for new parameter type definitions
    • Refined strategy specification structure

@gemini-code-assist

Copy link
Copy Markdown

Warning

You have reached your daily quota limit. Please wait up to 24 hours and I will start processing your requests again!

@diw-zw diw-zw changed the title 0506 tpe feat(cli): enhancements to the auto-benchmark search algorithms May 8, 2026

@Syspretor Syspretor left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

/lgtm

@Syspretor
Syspretor merged commit dc2044d into sgl-project:main May 8, 2026
10 checks passed
Syspretor pushed a commit that referenced this pull request May 8, 2026
* feat(cli): auto-benchmark add ui

* fix: parallel coordinates - categorical axis

* parse slaPass info from constraints

* feat(ui): enhance parameter comparison with categorical scatter chart and log scale

---------

Co-authored-by: 照微 <wd528567@alibaba-inc.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants