AWS、SageMaker Feature Storeに特徴量単位の書き込み機能を追加
AWS announced an UpdateRecord API for Amazon SageMaker Feature Store, allowing selected feature values to be changed without rewriting an entire record.
ひとことで言うと
AWSはAmazon SageMaker Feature Storeについて何を発表しましたか?
AWS announced UpdateRecord, a new Amazon SageMaker Feature Store API for partial record updates. Customers can change one or more feature values in a single call, while unmodified features remain unchanged. AWS says the operation applies atomically and is available in both Standard and In-Memory online store tiers.
要点
- AWS introduced the UpdateRecord API for Amazon SageMaker Feature Store.
- このAPIを使えば、アプリケーションはレコード全体を読み込んだり書き換えたりすることなく、1つ以上の機能値を更新できます。
- AWS says updates are applied atomically and features omitted from a request are preserved.
- この機能は、DynamoDBを基盤とする標準オンラインストア層と、ElastiCacheを基盤とするインメモリオンラインストア層で利用できます。
AWSは、新しいUpdateRecord APIを通じて、Amazon SageMaker Feature Storeで機能単位の書き込みが可能になったと発表した。同サービスは、機械学習の特徴量を保存、共有、管理するためのフルマネージド型リポジトリで、AWSは特徴量を、モデルの訓練や予測の生成に用いる加工済みデータと説明している。
UpdateRecordを使えば、アプリケーションは1回の呼び出しで1つ以上の特徴量の値を更新でき、レコード全体を読み取ったり書き換えたりする必要がない。AWSによると、Feature Storeは要求された変更を既存レコードに対してアトミックに適用し、リクエストに含まれない特徴量は変更されない。
レコード全体の更新に代わる仕組み
このAPIが登場する前は、1つの特徴量を変更するだけでも、GetRecordとPutRecordを使った読み取り・変更・書き込みの処理が必要だった。アプリケーションはまずレコード全体を読み取り、アプリケーションコードで新しい値を反映したうえで、レコード全体を書き戻していた。
AWSは例として、不正検知スコアリングのパイプラインを挙げている。顧客のrisk_scoreを更新するには、パイプラインがレコード内のすべての特徴量を取得してから、更新後のレコードを書き込まなければならなかった。
AWSによると、この処理は更新のたびに遅延を生じさせ、追加の読み取りキャパシティーを消費していた。また、同じレコード内の異なる特徴量を並行して更新するパイプライン同士で、競合状態が発生する可能性もあったという。最悪の場合、一方のパイプラインの書き込みが別のパイプラインの書き込みを気付かないうちに上書きすることがあり、AWSはこれを更新の消失(lost update)問題と呼んでいる。
AWSによると、追加のGetRecordリクエストにはRead Capacity Unitsの料金も発生する。頻繁に更新される幅広い特徴量グループを運用する顧客にとって、こうしたコストは積み重なる可能性があるという。
新しいAPIの仕組み
クライアントアプリケーションは、変更した特徴量だけを含むUpdateRecordリクエストを送信する。AWSによると、Feature StoreサービスはAWS Identity and Access Managementの権限を検証し、古い書き込みを拒否するためにEventTimeの順序を確認したうえで、アトミックなマージを実行する。
部分的な書き込みはオンラインストアに適用される。AWSによると、完全なレコードのスナップショットがオフラインストアに自動的に複製されるため、訓練用データセットの正確性が保たれる。発表では、クリックストリーム、購入、スコアリングなど複数のパイプラインが、調整なしに同じレコードへそれぞれの特徴量を独立して書き込めるとしている。
UpdateRecordは、Amazon SageMaker Feature Storeのオンラインストアに用意された2つのティア、Amazon DynamoDBを基盤とするStandardティアと、Amazon ElastiCacheを基盤とするIn-Memoryティアの双方で利用できる。
出典:AWS「Amazon SageMaker Feature Store introduces UpdateRecord for feature-level writes」(2026年9月8日公開)
よくある質問
- What is UpdateRecord?
- UpdateRecordは、レコード全体を書き換えることなく、既存のレコード内にある1つ以上の特徴量の値を変更するためのAmazon SageMaker Feature Store APIです。
- Which features are changed?
- アプリケーションは変更したい機能のみを指定します。AWSによれば、リクエストに含まれない機能はそのまま維持されます。
- Which Feature Store tiers support the API?
- AWSによると、UpdateRecordはAmazon DynamoDBを基盤とするStandardオンラインストア階層と、Amazon ElastiCacheを基盤とするIn-Memoryオンラインストア階層で利用できる。
- What process did UpdateRecord replace?
- 従来、1つのフィールドを変更するには、GetRecordでレコード全体を読み込み、アプリケーションコードで値をマージしたうえで、PutRecordでレコード全体を書き込む必要がありました。
出典
- Amazon SageMaker Feature Store introduces UpdateRecord for feature-level writes | Artificial Intelligence — Amazon Web Services (AWS)